检查DataFrame中的第n个值是否等于字符串中的第n个字符

Question

我有一个 df：

df =
     c1  c2   c3   c4  c5
  0  K   6    nan  Y   V
  1  H   nan  g    5   nan
  2  U   B    g    Y   L

还有一个字符串

s = 'HKg5'

我想返回 s[0]=c1 的值，s[1]=c2 的值，..... + 在某些情况下 s[i]=nan 的行。

例如上面df中的第1行与字符串匹配

    row 1=
           c1  c2   c3   c4  c5
        1  H   nan  g    5   nan
                                                match=True,   regardless of s[1,4]=nan
     s   = H   K    g    5

而且字符串长度是动态的，所以我的 df cols 高于 c10

我正在使用 df.apply，但我无法弄清楚。我想写一个函数传给df.apply，同时传递字符串。

感谢您的帮助！

克里斯的回答输出

  df=  
        c1  c2  c3  c4  c5 
     0  K   6  NaN  Y   V
     1  H  NaN  g   5  NaN
     2  U   B   g   Y   L

  s = 'HKg5'
  s1 = pd.Series(list(s), index=[f'c{x+1}' for x in range(len(s))])
  df.loc[((df == s1) | (df.isna())).all(1)]

输出

  `c1  c2  c3  c4  c5`

Answer 1

从您的字符串创建一个助手Series 并使用布尔逻辑进行过滤：

s1 = pd.Series(list(s), index=[f'c{x+1}' for x in range(len(s))])

# print(s1)    
# c1    H
# c2    K
# c3    g
# c4    5
# dtype: object

逻辑是 df 等于 (==) 这个值 OR (|) 是 nan (isna)
沿轴 1 使用 all 返回所有值为 True 的行

df.loc[((df == s1) | (df.isna())).all(1)]

[出]

  c1   c2 c3 c4   c5
1  H  NaN  g  5  NaN

---

所以，作为一个函数，你可以这样做：

def df_match_string(frame, string):
    s1 = pd.Series(list(string), index=[f'c{x+1}' for x in range(len(string))])
    return ((frame == s1) | (frame.isna())).all(1)

df_match_string(df, s)

[出]

0    False
1     True
2    False
dtype: bool

---

更新

我无法使用提供的示例重现您的问题。我的猜测是您的 DataFrame 中的某些值可能有前导/尾随空格？

在尝试上述解决方案之前，请尝试以下预处理步骤：

for col in df:
    df[col] = df[col].str.strip()