我试图导出一个numpy数组到不同的csv文件使用一个函数和基于 数据数组每行的第二个值。我的目标是在每行的第二个值相等时将数据导出到同一个文件。到目前为止,我可以创建不同的文件,但我也导出所有数据到每个文件。问题的第二部分是,我的数组中没有第二个值的最大选项数。这是我完成的代码:
a = np.array([(2,"Ana",9),(5,"Maria",4),(6,"Joao",3),
(1,"Ana",4)])
export_path = r"c:"
def export(array_values):
names = []
for i in xrange(len(array_values)):
names.append(array_values[i][1])
names = sorted(set(names))
for i in xrange(len(array_values)):
for j in xrange(len(names)):
if array_values[i][1] == names[j]:
name = "..."
export_file_path = os.path.join(export_path,name + ".csv")
myfile = open(export_file_path, 'wb')
wr = csv.writer(myfile, quoting=csv.QUOTE_NONE)
wr.writerows(array_values)
export(a)
事先谢谢你的帮助。 伊沃
首先,确定需要多少文件:
现在,您可以在组上循环,提取相应的行,然后保存它们:
你知道吗itertools.groupby组()使这变得容易。你知道吗
好吧:
应该生成3个文件:即
ana.txt
(由于名称已经成为文件名,我们可以从最终的csv文件中删除它):相关问题 更多 >
编程相关推荐