在Python中將列表轉換為數組dtype = int64

Question

我可以轉換此列表：

[0, 1, 2, 3, 4, 5, array([6, 9], dtype=int64), 7, 8, array([6, 9], dtype=int64)]

到這個：

[array([0], dtype=int64), array([1], dtype=int64), array([2], dtype=int64), array([3], dtype=int64), array([4], dtype=int64), array([5], dtype=int64), array([6, 9], dtype=int64), array([7], dtype=int64), array([8], dtype=int64), array([6, 9], dtype=int64)]

Answer 1

FWIW，這是一個不尋常的數據結構。 鑒於此，也許這樣的事情會起作用：

>>> a = [0, 1, 2, 3, 4, 5, np.array([6, 9], dtype=np.int64), 7, 8, np.array([6, 9], dtype=np.int64)] 
>>> b = [np.atleast_1d(np.asarray(x,dtype=np.int64)) for x in a]

這使

>>> pprint(b)
[array([0], dtype=int64),
 array([1], dtype=int64),
 array([2], dtype=int64),
 array([3], dtype=int64),
 array([4], dtype=int64),
 array([5], dtype=int64),
 array([6, 9], dtype=int64),
 array([7], dtype=int64),
 array([8], dtype=int64),
 array([6, 9], dtype=int64)]

Answer 2

這不行嗎？

L = [0, 1, 2, 3, 4, 5, array([6, 9], dtype=int64), 7, 8, array([6, 9], dtype=int64)]
new_L = [array(x, dtype=int64) for x in L]

編輯：

如果要單例數組，則

new_new_L = [array(x, dtype=int64) if type(x)==ndarray else array([x], dtype=int64) for x in L]