将列表转换为dtype = int64的数组

Question

我有一个清单

y_test  = [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1]

type(y_test)
Out[597]: list

我想将其转换为dtype int64的数组。

y_test
Out[598]: array([1, 1, 1, ..., 1, 0, 1], dtype=int64)

type(y_test)
Out[599]: numpy.ndarray

y_test.shape
Out[600]: (25000,)

请注意结果的形状。 这是可以实现的吗？

Answer 1

函数np.array可以采用任何序列并将其转换为numpy数组。 对于您的情况，这将是

np.array(y_test)
Out: array([0, 0, 0, ..., 1, 1, 1])

您还可以根据需要指定dtype参数：

np.array(y_test, dtype=np.int64)
Out: array([0, 0, 0, ..., 1, 1, 1])