在df下面包含了时间分组整数。我正在尝试将它们转换为numpy数组。
import pandas as pd
import numpy as np
df = pd.DataFrame({
'Time' : [1,1,1,2,2,2,3,3,3],
'A' : [3, 4, 5, 2, 5, 6, 1, 6, 7],
'B' : [2, 4, 5, 2, 5, 5, 2, 6, 5],
})
GA = np.array(df.groupby(['Time'])['A'].apply(np.array))
预期的：
[[ 3 4 5]
[ 2 5 6]
[ 1 6 7]]
用于转换之前列出
np.array(df.groupby('Time')['A'].apply(list).tolist())
出：
array([[3, 4, 5],
[2, 5, 6],
[1, 6, 7]])
本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
我来说两句