根據熊貓中另一列的值選擇一個列范圍

Question

我的數據集看起來像這樣（第一行是標題）

0    1    2    3    4    5
1    3    4    6    2    3
3    8    9    3    2    4
2    2    3    2    1    2

我想基於列[5]選擇數據集的列范圍，例如：

1    3    4
3    8    9    3
2    2

我已經嘗試了以下方法，但是沒有用：

df.iloc[:,0:df['5'].values]

Answer 1

我們試試吧：

df.apply(lambda x: x[:x.iloc[5]], 1)

輸出：

     0    1    2    3
0  1.0  3.0  4.0  NaN
1  3.0  8.0  9.0  3.0
2  2.0  2.0  NaN  NaN

Answer 2

重新創建您的數據框

df=pd.DataFrame([x[:x[5]] for x in df.values]).fillna(0)
df
Out[184]: 
   0  1    2    3
0  1  3  4.0  0.0
1  3  8  9.0  3.0
2  2  2  0.0  0.0