Проверьте, имеет ли какая-либо строка те же значения, что и массив numpy - PullRequest
0 голосов
/ 11 февраля 2020

Я работаю с pandas.Dataframe, который выглядит следующим образом:

       A  B  C  D
index
    1  0  0  0  1
    2  1  0  0  1 
    3  ... 
    4  ... 
    ...

И я создаю numpy.arrays, которые имеют ту же форму, что и строка в этом кадре данных. Я хочу проверить, присутствует ли созданный мною массив внутри фрейма данных. В этом случае, например, мой массив будет выглядеть так, если он находится в кадре данных :

  a=  [0,0,0,1]

Это не , если он выглядит так :

b = [1,1,1,1]

Любая помощь, даже если это ссылка на правильный ответ, очень ценится, поскольку я просмотрел stackoverflow и, к счастью, я ничего не пропустил.

1 Ответ

1 голос
/ 11 февраля 2020
df = pd.DataFrame({'A':[0, 1, 0, 0],
                   'B':[0, 0, 1, 1],
                   'C':[0, 0, 0, 0],
                   'D':[1, 1, 0, 1]})


#    A  B  C  D
# 0  0  0  0  1
# 1  1  0  0  1
# 2  0  1  0  0
# 3  0  1  0  1 



>>> a = [0, 0, 0, 1]
>>> (df == a).all(axis=1).any()
True

>>> b = [1, 1, 1, 1]
>>> (df == b).all(axis=1).any()
False
...