pandas读取多个csv文件,如果csv文件名字包含tt,则把第2列放到一个字典里面,列名字为tt。如果csv文件名字包含ss,则把第2列放到一个字典里面,列名字为ss。最后把字典输出到一个新的csv黎曼
时间: 2024-03-09 20:45:23 浏览: 163
好的,可以按照以下步骤操作:
1. 导入pandas库
```python
import pandas as pd
```
2. 定义函数,读取csv文件并将指定列保存到字典中
```python
def read_csv(file):
if 'tt' in file:
df = pd.read_csv(file)
data = { 'tt': df.iloc[:, 1] }
elif 'ss' in file:
df = pd.read_csv(file)
data = { 'ss': df.iloc[:, 1] }
else:
data = {}
return data
```
3. 遍历文件夹,读取符合条件的csv文件并将数据保存到字典中
```python
import os
path = 'csv_files_folder' # 文件夹路径
data = {}
for file in os.listdir(path):
if file.endswith('.csv'):
file_path = os.path.join(path, file)
data.update(read_csv(file_path))
```
4. 将字典转换为DataFrame,并保存到新的csv文件中
```python
df = pd.DataFrame(data)
df.to_csv('output.csv', index=False)
```
完整代码如下:
```python
import pandas as pd
import os
def read_csv(file):
if 'tt' in file:
df = pd.read_csv(file)
data = { 'tt': df.iloc[:, 1] }
elif 'ss' in file:
df = pd.read_csv(file)
data = { 'ss': df.iloc[:, 1] }
else:
data = {}
return data
path = 'csv_files_folder' # 文件夹路径
data = {}
for file in os.listdir(path):
if file.endswith('.csv'):
file_path = os.path.join(path, file)
data.update(read_csv(file_path))
df = pd.DataFrame(data)
df.to_csv('output.csv', index=False)
```
阅读全文