在熊貓中獲取函數列表

我在熊貓中擁有數據框，並且我寫了一個函數來使用每一行中的信息來生成一個新列。我想要的結果是在一個列表格式：在熊貓中獲取函數列表

 A B C 
     3 4 1 
     4 2 5 

    def Computation(row): 
      if row['B'] >= 3: 
       return [s for s in range(row['C'],50)] 
      else: 
       return [s for s in range(row['C']+2,50)] 

    df['D'] = df.apply(Computation, axis = 1)

不過，我收到以下錯誤：

「無法播放從形狀（308）輸入數組形狀（9）」

你能告訴我如何解決這個問題嗎？

來源

2016-09-25 user36729

假設您從

In [25]: df = pd.DataFrame({'A': [3, 4], 'B': [4, 2], 'C': [1, 5]})

那麼至少有兩種方法可以做到這一點。

您可以在C柱兩次申請，但在B柱上開關：

In [26]: np.where(df.B >= 3, df.C.apply(lambda c: [s for s in range(c, 50)]), df.C.apply(lambda c: [s for s in range(c + 2, 50)])) 
Out[26]: 
array([ [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49], 
     [7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49]], dtype=object)

或者你可以將整個行和開關適用於每行B值：

In [27]: df.apply(lambda r: [s for s in range(r.C, 50)] if r.B >= 3 else [s for s in range(r.C + 2, 50)], axis=1) 
Out[27]: 
0 [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14... 
1 [7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, ...

請注意，退貨類型不同，但在每種情況下，仍可編寫

df['foo'] = <each one of the above options>

來源

2016-09-25 19:17:59

這太好了。但是如果範圍的起始點取決於每行中B的值，我該如何實現？ – user36729

@ user36729你能舉一個你想做什麼的具體例子嗎？ b和4c之間的範圍會是一個很好的例子嗎？ –

我在我的主要問題中這樣做了。 – user36729

在熊貓中獲取函數列表

回答

相關問題