如何在数据框架中使用.loc函数，借助索引打印特定用户的数据。

Question

我已经创建了一个数据框架

tmp = pd.DataFrame.from_dict(all_pred)
tmp_transpose = tmp.transpose()

tmp_transpose.head(3)

然后我把index标记为userId

tmp_transpose = tmp_transpose.rename_axis('userId')
tmp_transpose

我想获取一个用户的数据，因此我得到一个错误信息

result=tmp_transpose.loc[196] 

Answer 1

看来你的索引是字符串 "196"，而不是数量 196

result = tmp_transpose.loc["196"]

---

最小的工作示例

import pandas as pd

#tmp = pd.DataFrame({'192': [1,2,3], '196': [1,2,3]})
tmp = pd.DataFrame.from_dict({'192': [1,2,3], '196': [1,2,3]})
tmp_transpose = tmp.transpose()

#print(tmp_transpose.head(3))

tmp_transpose = tmp_transpose.rename_axis('userId')
#print(tmp_transpose)

print(tmp_transpose.loc['196'])  # OK
print(tmp_transpose.loc[196])    # error