为什么 vstack 会改变元素的类型？我该如何解决这个问题？

Question

我有一些列表，例如

list1 = ['hi',2,3,4]
list2 = ['hello', 7,1,8]
list3 = ['morning',7,2,1]

其中 'hi'、'hello' 和 'morning' 是字符串，其余的是数字。

然后我尝试将它们堆叠为：

matrix = np.vstack((list1,list2,list3))

但是数字的类型变成了字符串。特别是他们变成了numpy_str。

我该如何解决这个问题？我尝试更换项目，我尝试更改它们的类型，但没有任何效果

编辑

上面写错了！在我原来的问题中，第一个列表实际上是一个标题列表，例如

list1 = ['hi', 'number of hours', 'number of days', 'ideas']

所以第一列（在垂直堆叠的数组中）是一列字符串。其他列的第一个元素是字符串，然后是数字。

Answer 1

您可以使用Pandas DataFrames，它们允许异构数据：

>>> pandas.DataFrame([list1, list2, list3])

         0  1  2  3
0       hi  2  3  4
1    hello  7  1  8
2  morning  7  2  1

如果你想命名列，你也可以这样做：

pandas.DataFrame([list1, list2, list3], columns=list0)

        hi  nb_hours  nb_days  ideas
0       hi         2        3      4
1    hello         7        1      8
2  morning         7        2      1

Answer 2

由于数字可以写成字符串，但字符串不能写成数字，所以你的matrix 的所有元素都是字符串类型的。

如果你想要一个整数矩阵，你可以： 1- 提取与您的数字对应的子矩阵，然后将其映射为整数 2- 或者您可以直接从列表中仅提取数字并堆叠它们。

import numpy as np
list1 = ['hi',2,3,4]
list2 = ['hello', 7,1,8]
list3 = ['morning',7,2,1]

matrix = np.vstack((list1,list2,list3))

# First
m = map(np.int32,matrix[:,1:])
# [array([2, 3, 4], dtype=int32), array([7, 1, 8], dtype=int32), array([7, 2, 1], dtype=int32)]

# Second
m = np.vstack((list1[1:],list2[1:],list3[1:]))
# [[2 3 4] [7 1 8] [7 2 1]]

---

编辑（回复评论）

我会调用标题列表list0：

list0 = ['hi', 'nb_hours', 'nb_days', 'ideas']

基本上是一样的想法：

1- 全部堆叠然后提取子矩阵（这里我们既不取第一行也不取第一列：[1:,1:]）

matrix = np.vstack((list0,list1,list2,list3))
matrix_nb = map(np.int32,matrix[1:,1:])

2- 直接不要堆叠list0 和堆叠所有其他列表（除了它们的第一个元素[1:]）：

m = np.vstack((list1[1:],list2[1:],list3[1:]))