pandas数据框,统计某列数据与其他文件对应关系的个数
程序员文章站
2022-06-05 19:06:08
...
现在要解决的问题如下:
我们有一个数据的表
第7列有许多数字,并且是用逗号分隔的,数字又与第二个文件有一个对应的关系:
我们要得到第7列对应关系的统计,就是每一行的第7列a有多少个,b有多少个
好了,我给的解决方法如下:
#!/bin/python
#-*-coding:UTF-8-*-
import pandas as pd
import numpy as np
dfidspec = pd.read_table("one.txt")#这个是对应关系的文件
dfmgs = pd.read_table("two.txt",header = None)#这个是我们数据的表
def getlistnum(li):#这个函数就是要对列表的每个元素进行计数
set1 = set(li)
dict1 = {}
for item in set1:
dict1.update({item:li.count(item)})
return dict1
bigdict = dict(zip(dfidspec['ID'],dfidspec['class']))#获得一个关系的字典
dfmgs['indeo'] = 'a'#在读取的数据框新建一个字符列
for i in range(len(dfmgs.index)):#对每一行进行操作
spp = [bigdict[int(j)] for j in dfmgs.iloc[i, 6].split(',')]#对于第7列的格子中的每个数进行字典取值
sppnum = getlistnum(spp)
dfmgs.iloc[i, 7] = str(sppnum)
dfmgs.to_csv("three.txt",sep = '\t',index = False)
这个就可以得到想要的结果了: