Computer >> 컴퓨터 >  >> 프로그램 작성 >> Python

Python - Pandas DataFrame에서 열을 선택하는 방법

<시간/>

DataFrame에서 열을 선택하려면 대괄호를 사용하여 가져오기만 하면 됩니다. 예를 들어 대괄호 안에 선택할 열을 언급하면 ​​끝입니다.

dataFrame[‘ColumnName’]

먼저 필요한 라이브러리를 가져옵니다 -

import pandas as pd

이제 DataFrame을 만듭니다. 두 개의 열이 있습니다 -

dataFrame = pd.DataFrame(
   {
      "Car": ['BMW', 'Lexus', 'Audi', 'Mustang', 'Bentley', 'Jaguar'],
      "Units": [100, 150, 110, 80, 110, 90]
   }
)

단일 열만 선택하려면 아래와 같이 대괄호를 사용하여 열 이름을 언급합니다. 여기에서 열 이름은 'Car'입니다 -

dataFrame ['Car']

예시

다음은 코드입니다 -

import pandas as pd

# Create DataFrame
dataFrame = pd.DataFrame(
   {
      "Car": ['BMW', 'Lexus', 'Audi', 'Mustang', 'Bentley', 'Jaguar'],
      "Units": [100, 150, 110, 80, 110, 90]
   }
)

print"DataFrame ...\n",dataFrame

# selecting a column
print"\nSelecting and displaying only a single column = \n",dataFrame ['Car']

출력

이것은 다음과 같은 출력을 생성합니다 -

DataFrame ...
       Car   Units
0      BMW     100
1    Lexus     150
2     Audi     110
3  Mustang      80
4  Bentley     110
5   Jaguar      90

Selecting and displaying only a single column =
0      BMW
1    Lexus
2     Audi
3  Mustang
4  Bentley
5   Jaguar
Name: Car, dtype: object