我目前正在嘗試從 Dataframe 中的單元格中檢索值。我正在搜索數(shù)據(jù)框以查找與一行中的列中的值匹配的字符串，然后從另一列返回該行中的值。我的代碼如下所示：df:Fruit Valueapple 7.0banana 6.0orange 8.0lemon 3.0melon 2.0myList = ['apple', 'lemon']result = []for word in myList: result.append(df['Value'].loc[df['Fruit'] == word].values)print(result)打印語句輸出：[array([], dtype=float64), array([7.0]), array([], dtype=float64), array([3.0])]我想要的輸出只是一個浮點值數(shù)組：[7.0, 3.0]我將如何清理我的輸出以實現(xiàn)這一目標？我的最終目標是獲得數(shù)組的平均值。
1 回答
繁花不似錦
TA貢獻1851條經(jīng)驗 獲得超4個贊
使用Series.isinwithboolean indexing并轉(zhuǎn)換為列表：
print (df.loc[df['Fruit'].isin(myList), 'Value'].values.tolist())
[7.0, 3.0]
我的最終目標是獲得數(shù)組的平均值。
然后更好的是調(diào)用mean過濾Series：
val = df.loc[df['Fruit'].isin(myList), 'Value'].mean()
print (val)
5.0
添加回答
舉報
0/150
提交
取消