Question

当我使用以下内容时：

import pandas as pd
data = pd.read_csv('C:/Users/Z/OneDrive/Python/Exploratory Data/Aramark/ARMK.csv')
x = data.iloc[:,2]
y = pd.unique(x)
y.to_csv('yah.csv')

我收到以下错误：

AttributeError: 'numpy.ndarray' object has no attribute 'to_csv'

Answer 1

IIUC，从数据框开始：

df = pd.DataFrame({'a':[1,2,3,4,5,6],'b':['a','a','b','c','c','b']})

您可以使用以下内容获取列的唯一值：

g = df['b'].unique()

返回一个数组：

array(['a', 'b', 'c'], dtype=object)

将其保存为.csv文件我会将其转换为Series s：

In [22]: s = pd.Series(g)

In [23]: s
Out[23]: 
0    a
1    b
2    c
dtype: object

所以你可以轻松保存它：

In [24]: s.to_csv('file.csv')

希望有所帮助。

Answer 2

相当于np.unique的大熊猫是drop_duplicates method。

In [42]: x = pd.Series([1,2,1,3,2])

In [43]: y = x.drop_duplicates()

In [46]: y
Out[46]: 
0    1
1    2
3    3
dtype: int64

请注意，drop_duplicates会返回一个系列，因此您可以调用其to_csv方法：

import pandas as pd
data = pd.read_csv('C:/Users/Z/OneDrive/Python/Exploratory Data/Aramark/ARMK.csv')
x = data.iloc[:,2]
y = x.drop_duplicates()
y.to_csv('yah.csv')

To_CSV pandas列的唯一值

2 个答案: