python-3.x - 使用 Pandas 数据帧时，无法将存储为 excel 中的字符串的矩阵转换为 numpy 数组

Question

我很难用 pandas DataFrame 读取 excel 文件并将存储的矩阵转换为 numpy array .我认为问题的一部分是矩阵存储不当。但是，我无法控制电子表格，这就是它发送给我的方式。

例如，这是存储在单元格中的字符串

[[[ 0.        0.        0.107851]
  [ 0.        0.       -0.862809]]]

我与 DataFrame 一起阅读，并将每个单元格保存到一个变量中。然后我尝试将此特定变量转换为 np.array因为这些数字代表两组 x、y、z 坐标。

我试过 np.fromstring和 np.asarray无济于事。它会将字符串转换为一个 numpy 数组，但如果括号内仍为字符，这将是一个可怕的困惑。我试过使用 np.squeeze 去掉括号，但它说维度不是 1。

如果我使用 np.asarray(item._coord, dtype=float)然后它失败说它不能将字符串转换为浮点数。
ValueError: could not convert string to float: '[[[ 0. 0. 0.107851] [ 0. 0. -0.862809]]]'
有一个 '\n' 出现在它的中间，在两个列表之间。我用 df = df.replace(r'\n', ' ',regex=True)' to clean out the\n` 在数据转换尝试之前。

我被卡住了

Answer 1

使用自定义函数转换为 numpy array之后 read_excel :

a= np.array([[[ 0.,        0.,        0.107851],
              [ 0.,        0.,       -0.862809]]])
print (a)
[[[ 0.        0.        0.107851]
  [ 0.        0.       -0.862809]]]

df = pd.DataFrame({'col':[a,a,a]})
print (df)
                                               col
0  [[[0.0, 0.0, 0.107851], [0.0, 0.0, -0.862809]]]
1  [[[0.0, 0.0, 0.107851], [0.0, 0.0, -0.862809]]]
2  [[[0.0, 0.0, 0.107851], [0.0, 0.0, -0.862809]]]

df.to_excel('test.xlsx', index=False)

import re
import ast
import numpy as np

#https://stackoverflow.com/a/44323021
def str2array(s):
    # Remove space after [
    s=re.sub('\[ +', '[', s.strip())
    # Replace commas and spaces
    s=re.sub('[,\s]+', ', ', s)
    return np.array(ast.literal_eval(s))

df = pd.read_excel('test.xlsx')

df['col'] = df['col'].apply(str2array)
print (df)
                                               col
0  [[[0.0, 0.0, 0.107851], [0.0, 0.0, -0.862809]]]
1  [[[0.0, 0.0, 0.107851], [0.0, 0.0, -0.862809]]]
2  [[[0.0, 0.0, 0.107851], [0.0, 0.0, -0.862809]]]