使用python找到多个csv文件中相同元素,并知道这些元素的索引名称
时间: 2023-05-24 11:01:49 浏览: 207
您好,以下是使用Python查找CSV文件中相同元素并找到它们的索引名称的代码示例:
```
import csv
# 定义要查找的元素
element = "example"
# 定义一个字典来存储每个CSV文件及其包含指定元素的行号
results = {}
# 遍历所有CSV文件
for file_name in ["file1.csv", "file2.csv", "file3.csv"]:
with open(file_name, "r") as csv_file:
reader = csv.reader(csv_file)
header = next(reader)
for i, row in enumerate(reader):
if element in row:
# 如果找到指定元素,则将其文件名称和行号添加到结果字典中
if file_name not in results:
results[file_name] = []
results[file_name].append(header[row.index(element)])
# 打印结果
for file_name, headers in results.items():
print(f"{file_name}:{headers}")
```
此代码将查找名为“example”的元素,并在包含此元素的行中查找它。它将找到所有使用的CSV文件,并为包含此元素的每一行记录其索引名称,最终将每个文件的名称和所有包含指定元素的行的索引名称打印到控制台。
阅读全文