python - 有没有什么方法可以获取第一个元素并计算它在 pandas 中的列中出现的次数？

Question

我试图获取多列中的第一个元素并计算该值出现的次数。

A       B
1,2,3 23,4,5
2     54 
2     2

result
1  1
2  3
54 1
32 1

Answer 1

使用DataFrame.stack对于系列，则 Series.str.split通过索引选择第一个值，如有必要，转换为整数并按 Series.value_counts 计数，如果需要的话最后排序:

s = df.stack().str.split(',').str[0].astype(int).value_counts().sort_index()
print (s)
1     1
2     3
23    1
54    1
dtype: int64

如果需要2列DataFrame:

df1 = (df.stack()
         .str.split(',')
         .str[0]
         .astype(int)
         .value_counts()
         .sort_index()
         .rename_axis('result')
         .reset_index(name='counts'))
print (df1)
   result  counts
0       1       1
1       2       3
2      23       1
3      54       1