python - Pandas 列的 To_CSV 唯一值

Question

这个问题在这里已经有了答案:





Dump a NumPy array into a csv file

(11 个回答)


5年前关闭。




当我使用以下内容时:

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')

希望有帮助。