Python：从多维表创建一组矢量坐标或矩阵

Question

如果我有一个有两个坐标的表格，比如(A,B)，并且每个坐标条目都有一个值 C。例如：

      B_1   B_2    B_3  ...
A_1   C1    C2     C3   ...
A_2   C4    etc    ...  ...
A_3   ...
...

我需要将其转换为矩阵格式，理想情况下是[ [A_1, B_1], [A_1, B_2], ... ] 形式的数组数组和每个[A,B] 点对应的C 值数组[ C1, C2, ]。

我需要这个的原因是因为我实际上有一个 3D（可能后来是 4D）表，我需要这样做，以便我可以对 C 值进行线性插值。

在 numpy 或 pandas 中是否有任何包或方法可以做到这一点？

目前我有这张纸上的表格，遍历点 (A,B,C) 及其相关值的所有排列将是非常乏味的。把它写成表格然后尽可能地转换会容易得多。

谢谢。

Answer 1

你的问题很模糊。您是否希望 A 和 B 成为指向某个 C 的索引？还是 A 和 B 本身携带值？

根据我的一些猜测，也许这会有所帮助：

import numpy as np

# A values
A = np.arange(0, 8)
# B values
B = np.arange(16, 24)
# The corresponding C Values, where C[n, m] corresponds to A_n, B_m
C = np.arange(8**2).reshape(8, 8)

# Creating an 3x8x8 Array
ABC = np.meshgrid(A, B) + [C,]
ABC = np.array(ABC)

print ABC

您可能还想查看 itertools.combinations