如何将序列的值（类型是string）更改为int？

Question

所以基本上我有这个数据框，在这个数据框里有一个系列的“shape”具有独特的值[“collinder”，“circle”，“light”，“cigar”，“diamond”，“oval”…]我想把这些形状转换成数字，这样我就可以用这些来绘制散点图了。在

当每个独特的形状都有自己的“id”作为int时，有没有办法制作另一个系列？在

编辑：设法让它与熊猫因子化

Answer 1

尝试skleanLabelEncoder将categorical列转换为Numerical，然后就可以绘制它了

import pandas as pd
df = pd.DataFrame(['cylinder', 'circle', 'light', 'cigar', 'diamond', 'oval'])
df.columns = ['shape']
from sklearn.preprocessing import LabelEncoder
le = LabelEncoder()
le.fit_transform(df)
plt.scatter(df.index , df['shape'])