Python 删除数组中的括号

Question

我有一个包含许多数组的列表。

coef

[array([[1.72158862]]),
 array([[3.28338167]]),
 array([[3.28004542]]),
 array([[6.04194548]])]

将其放入数据框中给出：

azone = pd.DataFrame(
    {'zone': zone,
     'coef': coef
    })

    zone    coef
0   1   [[1.7215886175218464]]
1   2   [[3.283381665861124]]

我想知道是否有办法去掉括号。我尝试了 tolist() 但没有给我结果。

另外还有另一个列表：

value

[[array([8.46565297e-294, 1.63877641e-002]),
 array([1.46912451e-220, 2.44570170e-003]),
 array([3.80589351e-227, 4.41242801e-004])]

我只想保留第二个值。期望输出为：

   value
0  1.63877641e-002
1  2.44570170e-003
2  4.41242801e-004

Answer 1

使用拉威尔：

coef = [np.array([[1.72158862]]),
        np.array([[3.28338167]]),
        np.array([[3.28004542]]),
        np.array([[6.04194548]])]

coef = np.array(coef).ravel()

print(coef)

array([1.72158862, 3.28338167, 3.28004542, 6.04194548])

此外，如果您不打算修改返回的一维数组，我建议您使用 numpy.ravel，因为它不会复制数组，而只是返回数组的视图，这要快得多比 numpy.flatten

Answer 2

您可以使用 NumPy 的flatten方法从多维数组列表中提取一维数组。例如：

coef = [np.array([[1.72158862]]),
        np.array([[3.28338167]]),
        np.array([[3.28004542]]),
        np.array([[6.04194548]])]

coef = np.array(coef).flatten()

print(coef)

array([1.72158862, 3.28338167, 3.28004542, 6.04194548])

由于 NumPy 数组是 Pandas 数据帧的基础，因此您会发现 Pandascoef系列现在是 dtypefloat并且仅包含标量。