python批量提取csv文件的特定列
时间: 2023-10-06 07:10:11 浏览: 121
可以使用Python中的pandas库来批量提取CSV文件的特定列。以下是一个示例代码:
```python
import pandas as pd
import glob
# 获取所有CSV文件的文件名
csv_files = glob.glob('*.csv')
# 遍历每个CSV文件并提取特定列
for file in csv_files:
# 读取CSV文件
df = pd.read_csv(file)
# 提取特定列,例如第2列和第4列
specific_cols = [2, 4]
df = df.iloc[:, specific_cols]
# 将提取后的数据保存为新的CSV文件
new_file_name = 'specific_cols_' + file
df.to_csv(new_file_name, index=False)
```
这段代码将读取所有CSV文件,并提取第2列和第4列,然后将提取后的数据保存为新的CSV文件,文件名以“specific_cols_”开头。你可以根据需要修改代码来提取不同的列。
相关问题
python批量提取csv文件中的特定几行
可以使用Python中的pandas库来批量提取CSV文件中的特定几行。以下是一个示例代码:
```python
import pandas as pd
import glob
# 获取所有CSV文件的文件名
csv_files = glob.glob('*.csv')
# 遍历每个CSV文件并提取特定几行
for file in csv_files:
# 读取CSV文件
df = pd.read_csv(file)
# 提取特定几行,例如第2行、第3行和第4行
specific_rows = [1, 2, 3]
df = df.iloc[specific_rows, :]
# 将提取后的数据保存为新的CSV文件
new_file_name = 'specific_rows_' + file
df.to_csv(new_file_name, index=False)
```
这段代码将读取所有CSV文件,并提取第2行、第3行和第4行,然后将提取后的数据保存为新的CSV文件,文件名以“specific_rows_”开头。你可以根据需要修改代码来提取不同的行。如果你想要提取特定的行名,可以使用`loc`方法来实现。
python批量提取csv文件中的特定名字的行
你可以使用pandas库中的`loc`方法来提取CSV文件中特定名字的行。以下是一个示例代码:
```python
import pandas as pd
import glob
# 获取所有CSV文件的文件名
csv_files = glob.glob('*.csv')
# 遍历每个CSV文件并提取特定名字的行
for file in csv_files:
# 读取CSV文件
df = pd.read_csv(file)
# 提取特定名字的行,例如名字为'Alice'和'Bob'
specific_rows = df.loc[df['Name'].isin(['Alice', 'Bob'])]
# 将提取后的数据保存为新的CSV文件
new_file_name = 'specific_rows_' + file
specific_rows.to_csv(new_file_name, index=False)
```
这段代码将读取所有CSV文件,并提取名字为'Alice'和'Bob'的行,然后将提取后的数据保存为新的CSV文件,文件名以“specific_rows_”开头。你可以根据需要修改代码来提取不同的行名。如果你想要提取特定行号,可以使用`iloc`方法来实现。
阅读全文