【发布时间】:2019-03-26 07:23:16
【问题描述】:
我在一个文件夹中有多个.csv 文件,它们具有相同的结构。
前两列包含两个值k1 和k2,第三列包含一个数值。
我想对这些文件进行迭代并将它们合并到一个新的csv 中,其中我将有K1 和K2 的前两列,然后是n 列,每个n 文件都有一个列。
这是我的解决方案:
import glob
import csv
import os
def get_data(filename):
'''function to read the data form the input csv file to use in the analysis'''
with open(filename, 'r') as f:
reader = csv.reader(f)
result = {tuple(row[:2]): row[2] for row in reader}
return result
path='mypath'
for infile in glob.glob(os.path.join(path, '*.csv')):
print ("Current File Being Processed is: " + infile)
#use split to seperate the path and name of the file
(PATH, FILENAME) = os.path.split(infile)
all_data=[]
#adds the data from the csv file to a blank list so it can be operated on
all_data.extend(get_data(infile))
result = {}
for dic in all_data:
for key in (dic.viewkeys() | result.keys()):
if key in dic: result.setdefault(key, []).append(dic[key])
else: result.setdefault(key, []).append(0)
with open('mypath', 'w') as csvfile:
writer = csv.DictWriter(csvfile, fieldnames=csv_columns)
writer.writeheader()
for data in result:
writer.writerow(data)
但它会返回
Traceback (most recent call last):
File "C:\Users\Desktop\p.py", line 26, in <module>
for key in (dic.viewkeys() | result.keys()):
AttributeError: 'tuple' object has no attribute 'viewkeys'
【问题讨论】:
-
你在用jupyter吗?
-
不,我正在研究 Python IDLE