파이썬 목록으로 Pandas 데이터 프레임 열 선택

내 스크립트 중 하나에서 열 이름 목록으로 데이터 프레임의 여러 열을 선택하고 있습니다. 다음 코드가 작동합니다.

data = df[lst]

목록의 모든 요소가 데이터 프레임에 포함되어 있으면 제대로 작동합니다. 그렇지 않은 경우 " '....'not in index"오류가 반환됩니다.

목록의 모든 요소가 데이터 프레임에 포함되지 않더라도 해당 목록에 포함 된 열 이름을 모두 선택할 수있는 가능성이 있습니까?

 

해결 방법

 

df = pd.DataFrame({'A':[1,2,3],
                   'B':[4,5,6],
                   'C':[7,8,9],
                   'D':[1,3,5],
                   'E':[5,3,6],
                   'F':[7,4,3]})

print (df)
   A  B  C  D  E  F
0  1  4  7  1  5  7
1  2  5  8  3  3  4
2  3  6  9  5  6  3

lst = ['A','R','B']

print (df.columns.intersection(lst))
Index(['A', 'B'], dtype='object')

data = df[df.columns.intersection(lst)]
print (data)
   A  B
0  1  4
1  2  5
2  3  6

data = df[np.intersect1d(df.columns, lst)]
print (data)
   A  B
0  1  4
1  2  5
2  3  6

 

참조 페이지 https://stackoverflow.com/questions/40636514