基于单独列中的值替换列的值

Question

我有一个熊猫数据帧，看起来像：

  ID  |   StateName   |   ZipCode
____________________________________  
  0          MD            20814     
  1                        90210   
  2          DC            20006   
  3                        05777   
  4                        12345

我有一个函数，它将根据ZipCode值填充StateName：

def FindZip(x):
    search = ZipcodeSearchEngine()
    zipcode = search.by_zipcode(x)
    return zipcode['State']

我想根据相应ZipCode的值来填充StateName列中的空白。我尝试过以下方法，但没有成功：

test['StateName'] = test['StateName'].apply(lambda x: FindZip(test['Zip_To_Use']) if x == "" else x)

基本上，我想将一个函数应用到一个不同于我试图更改的列的列。我会感激任何帮助！谢谢！你知道吗

Answer 1

我想出了一个不太“潘多拉”的解决办法。如果有人有想法的话，我还是很想看到一个更“pythonic”或“pandorable”的解决方案！我基本上创建了一个与DataFrame长度相同的新列表，遍历每一行，然后用新列表覆盖该列。你知道吗

state = [FindState(test['Zip_To_Use'].iloc[i]) if (test['StateName'].iloc[i] == "" and test['Zip_To_Use'].iloc[i] != "") 
         else test['StateName'].iloc[i] for i in range(len(test))]

在常规for循环中重述（为了可读性）：

state = []
for i in range(len(test)):
    if (test['StateName'].iloc[i] == "" and test['Zip_To_Use'].iloc[i] != ""):
        state.append(FindState(test['Zip_To-Use'].iloc[i]))
    else:
        state.append(test['StateName'].iloc[i])

然后用这个新列表重新分配列

test['StateName'] = state

如果你有更好的解决办法，请告诉我！你知道吗