python - Pandas:如果同一行上的另一个单元格包含字符串，则将变量设置为单元格中的值

Question

比方说我的 Pandas DataFrame包含以下内容:

Row    Firstname    Middlename Lastname
…
10     Roy         G.          Biv
11     Cnyder      M.          Uk
12     Pan         T.          One
…

如果 ["Lastname"] 中的一个单元格包含"Biv" ，我想设置Python变量first_name =至"Roy" .

我已经离开 Pandas 一段时间了，不确定实现这一目标的正确方法。我研究了组合 df.iloc 的方法/df.loc等等 df.at/df.where和str.contains但必须承认我有点迷失了将条件与设置变量放在一起的正确方法。

在(糟糕的，不正确的)伪代码中，我正在寻找:

first_name = df["Firstname"][np.where([df["Lastname"].str.contains("Biv") ...

Answer 1

如果您只想要标量值，那么您可以访问结果数组中的第一个值并分配:

firstname = df.loc[df['Lastname'].str.contains('Biv'), 'Firstname'][0]

请注意，实际上您应该检查它是否不为空:

if len(df.loc[df['Lastname'].str.contains('Biv'), 'Firstname']) > 0:
    firstname = df.loc[df['Lastname'].str.contains('Biv'), 'Firstname'][0]