Python: Pandas gelten Funktion: InvalidIndexError

Question

I apply auf meine Datenrahmen my_df wie unten bin mit:

my_df['column_C'] = my_df.apply(lambda x : 'hello' if x['column_B'] is None else x['column_B']) 

Ich möchte:

if x['column_B'] = None -> return 'hello' 
    if x['column_B'] != None -> return x['column_B'] 

Dann habe ich die folgenden Fehler:

<ipython-input-31-aa087c9a635e> in <lambda>(x) 
----> 1 my_df['column_C'] = my_df.apply(lambda x : 'hello' if x['column_B'] is None else x['column_B']) 

/usr/local/lib/python3.4/dist-packages/pandas/core/series.py in __getitem__(self, key) 
    599   key = com._apply_if_callable(key, self) 
    600   try: 
--> 601    result = self.index.get_value(self, key) 
    602 
    603    if not is_scalar(result): 

/usr/local/lib/python3.4/dist-packages/pandas/indexes/base.py in get_value(self, series, key) 
    2187    # python 3 
    2188    if is_scalar(key): # pragma: no cover 
-> 2189     raise IndexError(key) 
    2190    raise InvalidIndexError(key) 
    2191 

IndexError: ('column_B', 'occurred at index column_A') 

Weiß jemand, was ich hier falsch gemacht habe?

Answer 1

Sie müssen angeben, axis=1, um es auf jede Zeile, nicht jede Spalte anwenden. Lesen Sie die Dokumentation auf DataFrame.apply:

axis : {0 or 'index', 1 or 'columns'}, default 0 

* 0 or 'index': apply function to each column 
* 1 or 'columns': apply function to each row 

In Ihrem aktuellen Anruf, kann es nicht x['column_B'] finden, wenn es wirklich ein pd.Series verwenden, die column_A entspricht.

Also wenn Sie das folgende verwenden, wird es funktionieren.

my_df['column_C'] = my_df.apply(lambda x : 'hello' 
           if x['column_B'] is None 
           else x['column_B'], axis=1) 

Hinweis: wie im Kommentar oben erwähnt, DataFrame.fillna besser geeignet für diese Aufgabe ist.