从 Pandas DataFrame 中的列中获取唯一值

在 Pandas 中，要从 DataFrame 中的特定列中获取唯一值，可以使用 unique() 函数。使用 unique() 函数可以将 DataFrame 中的数据去重，只保留唯一值。

语法

DataFrame['column_name'].unique()

参数说明

• DataFrame: 要获取唯一值的 DataFrame。
• column_name: 要获取唯一值的列名。

返回值说明

unique() 函数返回一个由唯一值组成的 NumPy 数组。

代码示例

import pandas as pd

# 创建 DataFrame
data = {'fruit': ['apple', 'banana', 'orange', 'apple', 'banana']}
df = pd.DataFrame(data)

# 获取 'fruit' 列的唯一值
unique_fruits = df['fruit'].unique()

print(unique_fruits)

输出结果：

['apple' 'banana' 'orange']

以上代码演示了如何从 DataFrame 中的 'fruit' 列中获取唯一值。返回的结果是一个包含了唯一值的 NumPy 数组。这个功能对于对 Pandas DataFrame 中的数据进行分析和计算非常有用。