基于pandas数据框架的情节情感开发

Question

我正在做一个人脸检测和情感识别项目。我想在这样的图中描绘情感的发展。

Y轴是情绪，X轴取决于注册的情绪数量(0 -len(情绪))。对此有什么建议和想法吗？这就是我想要做的，但没有起作用，

​

Answer 1

你需要将'Happy‘等转换成数字，这就是为什么你会得到错误ValueError: the’t convert string to float：'Happy‘。您可以将所有独特的情绪放入一个列表中，并使用列表索引为每个情绪指定编号，为df中的新数字列指定编号。这里有一种方法：

df = pd.DataFrame({'Emotions': ['Happy', 'Disgust', 'Disgust', 'Happy']})

emotion_list = df['Emotions'].drop_duplicates().tolist()

for i, emotion in enumerate(emotion_list):
    df.loc[df['Emotions'] == emotion, 'emotion_as_number'] = i
    # or i + 1 if you want to start at 1 not 0

要获得

  Emotions  emotion_as_number
0    Happy                1.0
1  Disgust                2.0
2  Disgust                2.0
3    Happy                1.0