python - Python numpy数组(和Pandas索引)中不区分大小写的逻辑

我有一个熊猫系列，并进行了拼接：

print(s)
A            {B, A}
B     {B,  A   , E}
C          {B,  C}
D            {D, A}
E        {B, E,  C}
dtype: object

f = s.index
p = s.values

f现在是熊猫指数； p是一个numpy数组。然后，我删除空格。

我现在要“交叉检查”，看看每一行和每一列中都有哪些字母：

cross_check = (p[:, None] & [{x} for x in f]).astype(bool)
print(cross_check)

array([[ True,  True, False, False, False],
       [ True,  True, False, False,  True],
       [False,  True,  True, False, False],
       [ True, False, False,  True, False],
       [False,  True,  True, False,  True]], dtype=bool)

这很好，但是如果大小写不匹配（即第一行中的“ B”为“ b”）则失败。

如何执行逻辑且不区分大小写？谢谢！！

最佳答案

您可以使用列表推导通过set将upper转换为strip：

s = pd.Series([set(['B','A']),
               set(['B', ' a   ', 'E']),
               set(['B','  C']),
               set(['d','A']),
               set(['B','E', '  c'])], index=list('aBCDE'))
print (s)
a           {B, A}
B    {B, E,  a   }
C         {  C, B}
D           {d, A}
E      {  c, B, E}

f = s.index.str.upper().str.strip()
p = np.array([set([x.upper().strip() for x in item]) for item in s.values])
print (p)
[{'B', 'A'} {'B', 'E', 'A'} {'B', 'C'} {'D', 'A'} {'B', 'E', 'C'}]

cross_check = (p[:, None] & [{x} for x in f]).astype(bool)
print (cross_check)

[[ True  True False False False]
 [ True  True False False  True]
 [False  True  True False False]
 [ True False False  True False]
 [False  True  True False  True]]

对我来说，Zero解决方案也很好：

p = s.apply(lambda x: {v.strip().upper() for v in x})
print (p)
A       {B, A}
B    {B, E, A}
C       {B, C}
D       {D, A}
E    {B, E, C}
dtype: object

关于python - Python numpy数组(和Pandas索引)中不区分大小写的逻辑，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/46663158/