[PYTHON/PANDAS] DataFrame 클래스 : [] 연산자에서 조건식을 사용해 데이터 구하기

■ DataFrame 클래스의 [] 연산자에서 조건식을 사용해 데이터를 구하는 방법을 보여준다.

▶ main.py


import pandas as pd
import numpy  as np

dataFrame1 = pd.DataFrame(
    {
        "a" : np.random.randn(10),
        "b" : np.random.randn(10)
    }
)

dataFrame2 = dataFrame1[dataFrame1["a"] <= dataFrame1["b"]]

print(dataFrame2)

"""
          a         b
1 -0.184688  0.273896
2 -1.441498  0.736359
3 -1.427919  0.328332
4 -1.542995 -0.781992
5 -0.940297 -0.817988
7 -1.457501  2.540671
9  0.393672  1.654631
"""

import pandas as pd

import numpy as np

dataFrame1 = pd.DataFrame(

{

"a" : np.random.randn(10),

"b" : np.random.randn(10)

}

)

dataFrame2 = dataFrame1[dataFrame1["a"] <= dataFrame1["b"]]

print(dataFrame2)

"""

a b

1 -0.184688 0.273896

2 -1.441498 0.736359

3 -1.427919 0.328332

4 -1.542995 -0.781992

5 -0.940297 -0.817988

7 -1.457501 2.540671

9 0.393672 1.654631

"""

▶ requirements.txt


numpy==2.1.2
pandas==2.2.3
python-dateutil==2.9.0.post0
pytz==2024.2
six==1.16.0
tzdata==2024.2

numpy==2.1.2

pandas==2.2.3

python-dateutil==2.9.0.post0

pytz==2024.2

six==1.16.0

tzdata==2024.2

※ pip install pandas 명령을 실행한다.