python – 一个pandas列的To_CSV唯一值

发布时间：2020-12-20 11:50:34 所属栏目：Python 来源：网络整理

导读：参见英文答案 Dump a NumPy array into a csv file????????????????????????????????????10个当我使用以下内容时： import pandas as pddata = pd.read_csv('C:/Users/Z/OneDrive/Python/Exploratory Data/Aramark/ARMK.csv')x = data.iloc[:,2]y = pd.uniq

参见英文答案 > Dump a NumPy array into a csv file????????????????????????????????????10个
当我使用以下内容时：

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'

解决方法

IIUC,从数据框架开始：

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

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

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

返回一个数组：

array(['a','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')

希望有所帮助.

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!