pandas,读入多个csv,把它们的第2列取出来,为其添加两个元素,一个是A,一个是B。然后处理后的dataframe concat到新的csv,为新的csv指定新的index和columnn名字
时间: 2024-03-09 21:50:26 浏览: 49
基于pandas向csv添加新的行和列
可以按照以下步骤实现:
1. 使用pandas的`read_csv`函数读入多个csv文件,可以使用一个for循环遍历每个文件,并将读入的数据存储到一个列表中。例如:
```python
import pandas as pd
csv_list = ['file1.csv', 'file2.csv', 'file3.csv']
df_list = []
for csv_file in csv_list:
df = pd.read_csv(csv_file)
df_list.append(df)
```
2. 对每个读入的DataFrame,取出第二列,添加元素A和B,并将修改后的列重新赋值给原来的DataFrame。例如:
```python
for df in df_list:
df.iloc[:, 1] = df.iloc[:, 1].apply(lambda x: str(x) + '_A_B')
```
3. 使用pandas的`concat`函数将多个DataFrame合并成一个,并指定新的index和column名字。例如:
```python
new_df = pd.concat(df_list, ignore_index=True, axis=0)
new_df.index.name = 'Index'
new_df.columns = ['Column1', 'Column2']
```
4. 最后将新的DataFrame写入到一个csv文件中。例如:
```python
new_df.to_csv('new_file.csv')
```
阅读全文