创建名为＆＃34; Id＆＃34;的列基于行索引

Question

我想为我的数据框创建一个名为＆＃34; Id＆＃34;的新列。其中值是行索引+1。 我想像下面的例子：

   ID  Col1  ...
0  1   a     ...
1  2   b     ...
2  3   c     ...

Answer 1

您可以在索引中添加一个并将其分配到id列：

df = pd.DataFrame({"Col1": list("abc")})

df["id"] = df.index + 1

df
#Col1   id
#0  a    1
#1  b    2
#2  c    3

Answer 2

我将预测列表转换为数组，然后创建具有组总计的数据框并绘制该数据框

y_pred=model.predict(X_test)

#print(y_pred)
setosa=np.array([y_pred[i][0] for i in range(len(y_pred))])
versicolor=np.array([y_pred[i][1] for i in range(len(y_pred))])
virginica=np.array([y_pred[i][2] for i in range(len(y_pred))])

df2=pd.DataFrame({'setosa': [len(setosa[setosa>.5])],'versicolor': 
[len(versicolor[versicolor>.5])], 'virginica': [len(virginica[virginica>.5])] })
df2['id']=np.arange(0,1)
df2=df2.set_index('id')
df2.plot.bar()
plt.show()