dataframe转换为二维数组时带上index
要将DataFrame转换为带有索引的二维数组,可以使用DataFrame的values属性来获取值的二维数组,并使用index属性来获取索引的一维数组。然后将这两个数组合并成带有索引的二维数组。
下面是一个示例代码:
import pandas as pd
import numpy as np
# 创建一个DataFrame
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]}, index=['a', 'b', 'c'])
# 将DataFrame转换为二维数组
values = df.values
index = df.index.values
# 将values和index合并成带有索引的二维数组
result = np.column_stack([index, values])
print(result)
输出结果如下:
[['a' 1 4]
['b' 2 5]
['c' 3 6]]
这样,我们就得到了带有索引的二维数组。每一行的第一个元素为索引值,后面的元素为DataFrame中的数值
原文地址: https://www.cveoy.top/t/topic/iegI 著作权归作者所有。请勿转载和采集!