python如何从没有表头的csv文件中筛选出与txt文件相同的数据,并写入新的csv文件中
时间: 2024-02-15 19:03:27 浏览: 61
python将数据写入csv文件方法,说明
如果csv文件没有表头,可以使用Python内置的csv模块来读取和写入csv文件,以下是一个示例代码:
```python
import csv
# 读取csv文件
with open('data.csv', 'r') as f:
csv_reader = csv.reader(f)
data = [row for row in csv_reader]
# 读取txt文件
with open('conditions.txt', 'r') as f:
conditions = f.read().splitlines()
# 筛选出与txt文件中条件相同的数据
filtered_data = [row for row in data if row[0] in conditions]
# 将筛选结果写入新的csv文件中
with open('filtered_data.csv', 'w', newline='') as f:
csv_writer = csv.writer(f)
for row in filtered_data:
csv_writer.writerow(row)
```
其中,`data.csv`是csv文件的文件名,`conditions.txt`是txt文件的文件名,`filtered_data.csv`是筛选结果要存储的新的csv文件名。这里假设csv文件只有一列数据,所以筛选出的数据用`row[0]`进行比较。如果csv文件有多列数据,则需要根据具体情况进行修改。
阅读全文