Pandas kolumny swap w oparciu o stan

Question

Mam dataframe pandy tak:

Col1 Col2 Col3 
0 A  7  NaN 
1 B  16 NaN 
1 B  16 15 

Co chcę zrobić to zamienić COL2 z Col3 gdzie wartość Col3 jest NaN. Na podstawie innych stanowisk i odpowiedzi na SO, Mam ten kod do tej pory:

df[['Col2', 'Col3']] = df[['Col3', 'Col2']].where(df[['Col3']].isnull()) 

Ale to nie wydaje się działać prawidłowo i daje mi następujące:

Col1 Col2 Col3 
0 A  NaN NaN 
1 B  NaN NaN 
1 B  NaN NaN 

Czy jest coś, co ja może tu brakować?

Aktualizacja: My pożądane wyjście wygląda następująco:

Col1 Col2 Col3 
0 A  NaN 7 
1 B  NaN 16 
1 B  16 15 

Dzięki

Answer 1

Można użyć loc zrobić swap:

df.loc[df['Col3'].isnull(), ['Col2', 'Col3']] = df.loc[df['Col3'].isnull(), ['Col3', 'Col2']].values 

Zauważ, że .values wymagane jest, aby upewnić zamiana jest wykonana prawidłowo, w przeciwnym razie Panda spróbuje dopasować na podstawie nazw indeksów i kolumn oraz bez zamiany w mógłby wystąpić.

Można też po prostu przypisać każdy wiersz indywidualnie, jeśli czujesz, że kod jest czystsze:

null_idx = df['Col3'].isnull() 
df.loc[null_idx, 'Col3'] = df['Col2'] 
df.loc[null_idx, 'Col2'] = np.nan 

Wynikiem:

Col1 Col2 Col3 
0 A NaN 7.0 
1 B NaN 16.0 
2 B 16.0 15.0 

Answer 2

Spróbuj tego: (jej szybciej)

df["Col3"], df["Col2"] = np.where(df['Col3'].isnull(), [df["Col2"], df["Col3"]], [df["Col3"], df["Col2"] ]) 
df 

    Col1 Col2 Col3 
0 A NaN 7.0 
1 B NaN 16.0 
1 B 16.0 15.0 


    %timeit df.loc[df['Col3'].isnull(), ['Col2', 'Col3']] = df.loc[df['Col3'].isnull(), ['Col3', 'Col2']].values 
100 loops, best of 3: 2.68 ms per loop 


    %timeit df["Col3"], df["Col2"] = np.where(df['Col3'].isnull(), [df["Col2"], df["Col3"]], [df["Col3"], df["Col2"] ]) 
1000 loops, best of 3: 592 µs per loop