如何在循环过程中将提取的数据保存到不同的文件中?

时间:2019-04-26 13:44:35

标签: python-3.x numpy

输入的txt文件如下所示:

DATE ID ADDRESS
20161102 AB:2 xyzzzz
20161102 AB:2 fasafsdgs
20161102 AB:3 faasd
20161103 AB:3 fasfe
20161103 AB:4 ztwetwg

我想提取具有共同ID的单独txt文件中的数据。 例如:AB2_ID将包含仅与AB2 ID相关联的数据。

错误:

File "C:\Users\....\AppData\Local\Continuum\anaconda3\lib\site-packages\numpy\lib\npyio.py", line 1317, in savetxt
open(fname, 'wt').close()

OSError: [Errno 22] Invalid argument: 'D:\\Extracted_files\\AB:2_ID.txt'

代码:

for i in data.ID:
    subsetData = data[data['ID'] == i]
    file_path = r'D:\Extracted_files\%s_ID.txt' %i
    np.savetxt(file_path , subsetData.values, fmt = '%s')

0 个答案:

没有答案