[PYTHON/PANDAS] DataFrame 클래스 : [] 연산자를 사용해 특정 컬럼에서 특정 값들을 갖는 DataFrame 객체 구하기 1

■ DataFrame 클래스에서 [] 연산자를 사용해 특정 컬럼에서 특정 값들을 갖는 DataFrame 객체를 구하는 방법을 보여준다.

▶ main.py


import pandas as pd

dataFrame1 = pd.read_excel("titanic.xlsx", sheet_name = "passengers")

series = dataFrame1["Pclass"]

dataFrame2 = dataFrame1[series.isin([2, 3])]

print(dataFrame2)

"""
     PassengerId  Survived  Pclass                                     Name     Sex   Age  SibSp  Parch            Ticket     Fare Cabin Embarked
0              1         0       3                  Braund, Mr. Owen Harris    male  22.0      1      0         A/5 21171   7.2500   NaN        S
2              3         1       3                    Heikkinen, Miss Laina  female  26.0      0      0  STON/O2. 3101282   7.9250   NaN        S
4              5         0       3                 Allen, Mr. William Henry    male  35.0      0      0            373450   8.0500   NaN        S
5              6         0       3                         Moran, Mr. James    male   NaN      0      0            330877   8.4583   NaN        Q
7              8         0       3            Palsson, Master Gosta Leonard    male   2.0      3      1            349909  21.0750   NaN        S
..           ...       ...     ...                                      ...     ...   ...    ...    ...               ...      ...   ...      ...
884          885         0       3                   Sutehall, Mr. Henry Jr    male  25.0      0      0   SOTON/OQ 392076   7.0500   NaN        S
885          886         0       3     Rice, Mrs. William (Margaret Norton)  female  39.0      0      5            382652  29.1250   NaN        Q
886          887         0       2                    Montvila, Rev. Juozas    male  27.0      0      0            211536  13.0000   NaN        S
888          889         0       3  Johnston, Miss Catherine Helen "Carrie"  female   NaN      1      2        W./C. 6607  23.4500   NaN        S
890          891         0       3                      Dooley, Mr. Patrick    male  32.0      0      0            370376   7.7500   NaN        Q

[675 rows x 12 columns]
"""

import pandas as pd

dataFrame1 = pd.read_excel("titanic.xlsx", sheet_name = "passengers")

series = dataFrame1["Pclass"]

dataFrame2 = dataFrame1[series.isin([2, 3])]

print(dataFrame2)

"""

PassengerId Survived Pclass Name Sex Age SibSp Parch Ticket Fare Cabin Embarked

0 1 0 3 Braund, Mr. Owen Harris male 22.0 1 0 A/5 21171 7.2500 NaN S

2 3 1 3 Heikkinen, Miss Laina female 26.0 0 0 STON/O2. 3101282 7.9250 NaN S

4 5 0 3 Allen, Mr. William Henry male 35.0 0 0 373450 8.0500 NaN S

5 6 0 3 Moran, Mr. James male NaN 0 0 330877 8.4583 NaN Q

7 8 0 3 Palsson, Master Gosta Leonard male 2.0 3 1 349909 21.0750 NaN S

.. ... ... ... ... ... ... ... ... ... ... ... ...

884 885 0 3 Sutehall, Mr. Henry Jr male 25.0 0 0 SOTON/OQ 392076 7.0500 NaN S

885 886 0 3 Rice, Mrs. William (Margaret Norton) female 39.0 0 5 382652 29.1250 NaN Q

886 887 0 2 Montvila, Rev. Juozas male 27.0 0 0 211536 13.0000 NaN S

888 889 0 3 Johnston, Miss Catherine Helen "Carrie" female NaN 1 2 W./C. 6607 23.4500 NaN S

890 891 0 3 Dooley, Mr. Patrick male 32.0 0 0 370376 7.7500 NaN Q

[675 rows x 12 columns]

"""

▶ requirements.txt


defusedxml==0.7.1
et_xmlfile==2.0.0
numpy==2.1.2
odfpy==1.4.1
openpyxl==3.1.5
pandas==2.2.3
python-calamine==0.2.3
python-dateutil==2.9.0.post0
pytz==2024.2
pyxlsb==1.0.10
six==1.16.0
tzdata==2024.2
xlrd==2.0.1
XlsxWriter==3.2.0

defusedxml==0.7.1

et_xmlfile==2.0.0

numpy==2.1.2

odfpy==1.4.1

openpyxl==3.1.5

pandas==2.2.3

python-calamine==0.2.3

python-dateutil==2.9.0.post0

pytz==2024.2

pyxlsb==1.0.10

six==1.16.0

tzdata==2024.2

xlrd==2.0.1

XlsxWriter==3.2.0

※ pip install "pandas[excel]" 명령을 실행했다.

titanic.xlsx