python - 使用 Pandas 检查值是否存在或新的

Question

我有 2 个列表。

A = [1,2,3,5,8]

B = [2,3,4,5,6,7]

我想创建一个 pandas df 这样:

如果值只存在于A中，则在df中为该值追加状态为A，对于B类似，如果值同时存在于A和B中，则在df中追加A，B。例如:

values    status

1          A
2          A,B
3          A,B
4          B
5          A,B 
6          B
7          B
8          A

Answer 1

您可以使用一个简单的方法并应用来做一些

def find(x):
    if x in A and x in B:
        return 'A,B'
    elif x in A:
        return 'A'
    elif x in B:
        return 'B'     
    else: 
        return None
A = [1,2,3,5,8]
B = [2,3,4,5,6,7]
df = pd.DataFrame({'values':[1,2,3,4,5,6,7,8,9,10]})
df['status'] = df['values'].apply(find)

输出:

 values status0  1    A1  2  A,B2  3  A,B3  4    B4  5  A,B5  6    B6  7    B7  8    A8  9  None9  10 None