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

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

说：

这是我的密码：

def A（）：
def计数唯一（数据最终）：
返回数据\u final.nunique（dropna=False）
打印（计数唯一（数据最终））
()

是的，我挺过来了。结果是这样的

A 2

B 2

C 3

现在，我只想提取第二列，即，[2,2,3]作为列表。这将有助于我使用

column1=col\u name.repeat（）。

---

• 当我出错时， 对象对于所需数组太深 为什么呢

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

---

提前谢谢。

你就快拿到了

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

---

谢谢它成功了：）。。如何为该列表分配变量名。所以，我可以在以后的阶段使用它。现在。如果我想提取第一列，即[A，B，C]。我该怎么办？？

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