Python中判断某列是否只有两个唯一值

如果你需要判断DataFrame中的某列是否只有两个唯一值，可以使用以下代码：

if df[col].nunique() == 2:
    # do something

其中，df是一个DataFrame对象，col为该DataFrame中的某个列名。

需要注意的是，这里使用的是nunique()函数，而不是unique()函数。nunique()函数计算一个序列中的唯一值的数量，返回值为整型；而unique()函数返回序列中的唯一值组成的数组。

如果你真的需要使用unique()函数，可以使用以下代码：

if len(df[col].unique()) == 2:
    # do something

这里使用了len()函数，因为unique()函数返回的是一个数组，我们需要计算数组的长度，才能判断唯一值的数量是否为2。

需要注意的是，这里使用==符号进行比较，因为我们只关心唯一值的数量是否为2，不关心具体的唯一值是什么。如果你希望判断具体的唯一值，可以使用以下代码：

if set(df[col].unique()) == set(['value1', 'value2']):
    # do something

其中，['value1', 'value2']是一个包含两个唯一值的列表，使用set()函数将其转换为集合，并使用==符号进行比较。需要注意的是，这里使用了单引号将唯一值括起来，因为在Python中，字符串可以使用单引号或双引号进行表示，但是如果唯一值本身包含单引号或双引号，则需要使用另一种引号将唯一值括起来。

以上就是判断DataFrame中某一列是否只有两个唯一值的几种方法。希望对你有所帮助！