我想做一个火车数据集的列表

Question

假设我有一个数据集

  a       b         label 
  2    apple         4
  1    bin           5

我想把清单列成

[[apple,4], [bin,5]] 

Answer 1

尝试：

df.drop('a', axis=1).values.tolist()
[['apple', 4], ['bin', 5]]

Answer 2

import pandas as pd 
import numpy as np

df = pd.DataFrame([[2,'apple',4],[1,'bin',5]], columns = ['a','b','label'])
np.stack((df.b, df.label), axis = 1)

#op
array([['apple', 4],
   ['bin', 5]], dtype=object)