pandas相当于SELECT * FROM表WHERE column1 = column2

Question

什么是相当于'SELECT * FROM table WHERE column1 = column2'的pandas？

您有一个数据框，两列有值。您希望两列中的数字相同的所有行。那是什么代码？

dataframe:
column1   column2
a        b
b        a
c        c
d        d
a        b
a        b

The result I want:
column1   column2
c        c
d        d

谢谢。

Answer 1

在这种情况下，您将使用Pandas中的一些名为Masking

的东西

基本上，DataFrame [条件，在列上或整个数据框本身]返回条件为True的DataFrame。

import pandas as pd
import numpy as np

data = {'a':np.random.randint(0, 10, 100),
       'b':np.random.randint(0, 10, 100)}

df = pd.DataFrame(data)
df[df.a==df.b]