Блог

У меня есть два фрейма данных, где фрейм данных A содержит гораздо больше столбцов, чем фрейм данных B, что я хотел бы сделать, это отфильтровать фрейм данных A, используя фрейм данных B в качестве ссылки, и получить новый фрейм данных A с тем же количеством столбцов, что и фрейм данных A. Например:

 df_A = pd.DataFrame(np.random.randn(150, 17), columns=list('ABCDEFGHIJKLMONPQ'))
df_B = pd.DataFrame(np.random.randn(150, 8), columns=list('ABCDEFGH'))
 

Я хотел бы отфильтровать дополнительные столбцы df_A и иметь a df_A с теми же столбцами, которые df_B есть.

Чтобы df_A на выходе были столбцы ‘ABCDEFGH’

Используйте фильтр.

 df_A.filter(df_B.columns)
 

Или

 df_A[df_B.columns]