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

如何在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条回答 默认 最新

  • 关注

    这种不知道算不算简单了

    import pandas as pd
    import numpy as np
     
    df = pd.DataFrame()
    df[['A', 'B']] = pd.DataFrame(np.arange(10).reshape((5, 2)))
    
    # 变成字典,以A为keys
    df.set_index('A').T.to_dict('list')[4][0]
    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论
查看更多回答(2条)

报告相同问题?

问题事件

  • 已结题 (查看结题原因) 8月19日

悬赏问题

  • ¥15 R语言卸载之后无法重装,显示电脑存在下载某些较大二进制文件行为,怎么办
  • ¥15 java 的protected权限 ,问题在注释里
  • ¥15 这个是哪里有问题啊?
  • ¥15 关于#vue.js#的问题:修改用户信息功能图片无法回显,数据库中只存了一张图片(相关搜索:字符串)
  • ¥15 texstudio的问题,
  • ¥15 spaceclaim模型变灰色
  • ¥15 求一份华为esight平台V300R009C00SPC200这个型号的api接口文档
  • ¥15 字符串比较代码的漏洞
  • ¥15 欧拉系统opt目录空间使用100%
  • ¥15 ul做导航栏格式不对怎么改?