熊猫只从单元格返回值

时间:2019-04-02 11:55:03

标签: python pandas dataframe jupyter jupyter-lab

我目前正在尝试从数据框的单元格中检索值。我正在搜索一个数据框,以查找与一行中某一列中的值匹配的字符串,然后从另一列中返回该行中的值。

我的代码如下:

df:
Fruit   Value
apple     7.0
banana    6.0
orange    8.0
lemon     3.0
melon     2.0

myList = ['apple', 'lemon']

result = []

for word in myList:

    result.append(df['Value'].loc[df['Fruit'] == word].values)

print(result)

print语句正在输出:

[array([], dtype=float64), array([7.0]), array([], dtype=float64), array([3.0])]

我想要的输出只是一个浮点值数组:

[7.0, 3.0]

我将如何清理输出以实现此目标?我的最终目标是获得数组的平均值。

1 个答案:

答案 0 :(得分:1)

Series.isinboolean indexing一起使用并转换为列表:

print (df.loc[df['Fruit'].isin(myList), 'Value'].values.tolist())
[7.0, 3.0]
  

我的最终目标是获取数组的平均值。

那么最好调用过滤后的mean的{​​{1}}:

Series