【求助】数据处理/一维数据转三维数组

[AllenCoraline]

/pandas、numpy、数据处理
在https://github.com/RobinLu1209/STAG-GCN中，关于数据处理的代码楼主没发，联系不上，于是我决定做下数据处理的部分
求大佬帮忙！
分割时间1-4月，然后把第一列节点，第二第三列时间，第五列特征，整合成个三维数组

[陈年椰子]

df = pd.read_csv('data/data1.csv')
print([df['org'].to_list(),df['op_time'].to_list(),df['t2'].to_list()])

[陈年椰子]

看起来是用 pd.read_csv 读文本，分隔符为 逗号 （按逗号分隔，你说的第2，3列，其实就是1列） 取第1，2，4列，转成数组输出。

[AllenCoraline]

大佬能否求个取第1，2，4列转成数组输出的代码