我想将列表写入csv,当尝试这样做时,我收到以下错误
out.writerows(fin_city_ids)
_csv.Error: iterable expected, not numpy.int64
我的代码如下
org_id.append([pol_id,bldest_id])
fin_ids=list(org_city_id['org_id'].unique())
print(fin_ids)
out = csv.writer(open("D:/dataset/fin_ids.csv","w"), delimiter='|')
out.writerows(fin_ids)
下面是fin_ids
的输出[1002774, 0, 1000702, 1000339, 1001620, 1000710, 1000202, 1003143, 147897, 31018, 1001502, 1002812, 1003026, 1003280, 1003289, 1002714, 133191, 5252218, 6007821, 1002632]
Org_id是一个包含重复ID的dataFrame .fin_ids是一个包含ID的唯一值的列表。fin ID是一个从数据帧Org_id派生的唯一ID列表。
所需的输出是一个CSV,其中所有值位于单独的行中,因为我稍后将把数据加载到sql表中。
您可以通过多种方式完成此操作。但是,如果您希望从csv
模块中获取writerows
,那么您必须首先将列表fin_ids
转换为列表序列:
fin_ids = [1002774, 0, 1000702, 1000339,
1001620, 1000710, 1000202, 1003143, 147897,
31018, 1001502, 1002812, 1003026, 1003280,
1003289, 1002714, 133191, 5252218, 6007821, 1002632]
outfile = open('D:/dataset/fin_ids.csv','w')
out = csv.writer(outfile)
out.writerows(map(lambda x: [x], fin_ids))
outfile.close()
另一种方法是使用熊猫Series
中的.to_csv()
方法。因为您从一个数据框开始,所以您可以这样做:
org_city_id['org_id'].unique().to_csv("D:/dataset/fin_ids.csv", index=False)
这两个都应该生成一个包含以下数据的csv文件:
1002774
0
1000702
1000339
1001620
1000710
1000202
1003143
147897
31018
1001502
1002812
1003026
1003280
1003289
1002714
133191
5252218
6007821
1002632