qq_38411575
2021-04-08 14:12
采纳率: 77.8%
浏览 1.1k
已结题

如何在Dataframe中,根据某列的值取另外一列的某个值的标量?

我现在用的是最原始的方法,代码如下:

import pandas as pd
import numpy as np

df = pd.DataFrame()
df[['A', 'B']] = pd.DataFrame(np.arange(10).reshape((5, 2)))
df = df.loc[df['A'] == 4, 'B'].values[0]
print(df)

结果

5

请问,这行能再简化么?

df = df.loc[df['A'] == 4, 'B'].values[0]

大牛回答的:

方法1:如果一直能匹配使用Series.to_numpy()代替Series.values

https://pandas.pydata.org/pandas-docs/stable/reference/api/pandas.Series.to_numpy.html

df = df.loc[df['A'] == 4, 'B'].to_numpy()[0]

方法2:如果可能不匹配,使用next和iter1,因为使用.loc可能出错

a = next(iter(df.loc[df['A'] == 4, 'B']), 'no match')
print (a)
5

a = next(iter(df.loc[df['A'] == 1000, 'B']), 'no match')
print (a)
no match

方法3:

df1=df.loc[df['A'] == 4, 'B'].item()
  • 写回答
  • 好问题 提建议
  • 追加酬金
  • 关注问题
  • 邀请回答

3条回答 默认 最新

相关推荐 更多相似问题