python - 如何从函数值中提取特定列？

Question

所以，我有一个分类数据框。我必须提取 df 中所有列的唯一值。

说:

A B C

a c b 

b c t

a d h

a d b

这是我的代码:

def A():

    def count_unique(data_final):

        return data_final.nunique(dropna=False)

    print(count_unique(data_final))

A()

是的，我通过这个得到了它。结果看起来像这样

A 2

B 2

C 3

现在，我只想提取第二列，即 [2,2,3] 作为列表。这将帮助我使用 column1=col_name.repeat()。

<小时/>

• 我遇到错误的地方， 对象对于所需数组来说太深 这是为什么？

• 如何提取第二列并将其转换为列表？

<小时/>

提前致谢。

Answer 1

你就快明白了

def count_unique(data_final):
    return data_final.nunique(dropna=False).values.tolist()