Python Pandas : Convert ".value_counts" output to dataframe

Question

您好，我想获取数据帧的唯一值的计数。 count_values 实现了这一点，但是我想在其他地方使用它的输出。如何将 .count_values 输出转换为 pandas 数据框。这是一个示例代码:

import pandas as pd
df = pd.DataFrame({'a':[1, 1, 2, 2, 2]})
value_counts = df['a'].value_counts(dropna=True, sort=True)
print(value_counts)
print(type(value_counts))

输出是:

2    3
1    2
Name: a, dtype: int64
<class 'pandas.core.series.Series'>

我需要的是这样的数据框:

unique_values  counts
2              3
1              2

谢谢。

Answer 1

使用 rename_axis来自索引的列名和 reset_index :

df = df.value_counts().rename_axis('unique_values').reset_index(name='counts')
print (df)
   unique_values  counts
0              2       3
1              1       2

或者如果需要一列 DataFrame 使用 Series.to_frame :

df = df.value_counts().rename_axis('unique_values').to_frame('counts')
print (df)
               counts
unique_values        
2                   3
1                   2