Pandas Dataframe에서 여러 열 선택
Manav Narula
2023년1월30일
Pandas DataFrame에서 여러 열의 데이터를 추출 할 때 주로 Dataframe을 2 차원 배열처럼 취급하기 때문에 문제가 발생할 수 있습니다. DataFrame에서 여러 열을 선택하려면 열 이름 목록을getitem 구문 ([])에 전달하거나 Pandas에서 제공하는iloc()및loc()메서드를 사용하여 기본 색인 생성 방법을 사용할 수 있습니다. 도서관. 이 자습서에서는 다음 DataFrame에서 여러 열을 선택합니다.
DataFrame의 예:
import pandas as pd
import numpy as np
df = pd.DataFrame(np.random.rand(4, 4), columns=["a", "b", "c", "d"])
print(df)
출력:
a b c d
0 0.255086 0.282203 0.342223 0.263599
1 0.744271 0.591687 0.861554 0.871859
2 0.420066 0.713664 0.770193 0.207427
3 0.014447 0.352515 0.535801 0.119759
__getitem__ 구문 ([])을 사용하여 여러 열 선택
추출 할 열 이름을 목록에 저장 한 다음[]에 전달하면 DataFrame에서 여러 열을 선택할 수 있습니다. 다음 코드는 이전에 표시된 DataFrame에서 a및 c열을 선택하는 방법을 설명합니다.
import pandas as pd
import numpy as np
df = pd.DataFrame(np.random.rand(4, 4), columns=["a", "b", "c", "d"])
print(df[["a", "c"]])
출력:
a c
0 0.255086 0.342223
1 0.744271 0.861554
2 0.420066 0.770193
3 0.014447 0.535801
iloc()및 loc() 메서드를 사용하여 Pandas에서 여러 열 선택
iloc()및loc()메서드를 사용하여 여러 열을 선택할 수도 있습니다.
열 인덱스를 사용하여 추출하려면 아래 예제와 같이iloc()을 사용할 수 있습니다.
import pandas as pd
import numpy as np
df = pd.DataFrame(np.random.rand(4, 4), columns=["a", "b", "c", "d"])
print(df.iloc[:, [0, 2]])
출력:
a c
0 0.255086 0.342223
1 0.744271 0.861554
2 0.420066 0.770193
3 0.014447 0.535801
마찬가지로 아래와 같이 이름을 사용하여 열을 선택하려는 경우loc()을 사용할 수 있습니다.
import pandas as pd
import numpy as np
df = pd.DataFrame(np.random.rand(4, 4), columns=["a", "b", "c", "d"])
print(df.loc[:, ["a", "c"]])
출력:
a c
0 0.255086 0.342223
1 0.744271 0.861554
2 0.420066 0.770193
3 0.014447 0.535801
작가: Manav Narula
Manav is a IT Professional who has a lot of experience as a core developer in many live projects. He is an avid learner who enjoys learning new things and sharing his findings whenever possible.
LinkedIn