如何使用Python将数据插入CSV文件列中?
时间: 2024-12-16 10:23:36 浏览: 4
Python将一个CSV文件里的数据追加到另一个CSV文件的方法
要将数据插入CSV文件的特定列中,可以使用pandas库[^2]。以下是一个示例:
```python
import pandas as pd
# 假设我们有一个字典dic_data,包含了要插入的数据
dic_data = {'报考人数': [100, 200, 300], '录取人数': [50, 75, 100], '年龄': [20, 25, 30], '毕业人数': [80, 90, 100]}
# 将字典转换为DataFrame
df_data = pd.DataFrame(dic_data)
# 如果你的现有CSV文件名为'test1.csv',并且你想追加这些新数据到现有的某一列(假设是'报考人数'),你可以这样做:
# 加载现有CSV文件
existing_df = pd.read_csv('test1.csv')
# 添加新数据的列名,如果它不存在于现有文件中
if '报考人数' not in existing_df.columns:
existing_df['报考人数'] = None
# 追加数据到已有的'报考人数'列
existing_df['报考人数'].extend(df_data['报考人数'])
# 写入更新后的DataFrame到CSV文件
existing_df.to_csv('test1.csv', index=False, mode='a', header=existing_df.columns.tolist() if existing_df.iloc[0].isnull().all() else False)
```
在这个示例中,我们首先检查新数据的列名是否存在于CSV文件中,然后使用`extend()`方法追加新的数据,最后以追加模式(`mode='a'`)写回CSV文件。
阅读全文