Wählen Sie Spalten in DF gemäß den Spaltennamen aus ⇐ Python

[Anonymous]

Ich habe eine df_source. df_new.
Das resultierende df_new kann entweder zwei Spalten oder vier Spalten enthalten.

Code: Select all

df_new = df_source[["A_1", "B_1"]]
if 'A_2' in df_new.columns:
# Also add A_2, B_2 columns in df_new, but I don't know how to add the columns to the created df_new.
< /code>
statt < /p>
df_new = df_source[["A_1", "B_1"]]
if 'A_2' in df_new.columns:
df_new = df_source[["A_1", "B_1", "A_2", "B_2"]]