提取csv文件中特定列的信息存入一个新csv文件中: 现有两个csv文件A和B,均含有列名为日期、航班号、WTC、跑道、机位的五列数据,读取这些内容,并根据先按航班号排序、每个航班再按日期从小到大排序的方式存放于一个新csv文件中
时间: 2024-04-29 21:26:40 浏览: 89
python实现对csv文件的读取和对第5列的排序
ChitGPT:
你好,我可以帮你解决提取csv文件中特定列的信息存入一个新csv文件中的问题。首先,您可以使用Python中的pandas库来读取CSV文件,然后使用DataFrame中的sort_values()函数对数据按航班号和日期进行排序。接着,您可以使用drop()函数将自己不需要的列删除,然后使用to_csv()函数将结果写入新的CSV文件中。下面是Python代码示例:
```python
import pandas as pd
# 读取CSV文件
df_a = pd.read_csv('A.csv')
df_b = pd.read_csv('B.csv')
# 将两个文件合并
df = pd.concat([df_a, df_b])
# 根据航班号和日期排序
df = df.sort_values(by=['航班号', '日期'])
# 仅保留需要的列
df = df[['日期', '航班号', 'WTC', '跑道', '机位']]
# 将结果写入新的CSV文件
df.to_csv('new.csv', index=False, encoding='utf-8-sig')
```
此代码将从文件A.csv和B.csv读取数据,将它们合并,并按航班号和日期排序。然后,仅保留需要的列,并将结果写入新的CSV文件new.csv中。
希望这可以帮助您解决您的问题!
阅读全文