python - 我想创建一个基于 2 个 np.ranges 的 pandas DF

Question

我想创建一个基于 2 个 np.arrays 的具有 2 列的 pandas DF。

最后它应该看起来像一个溶解的 x-y 矩阵，因为我必须测试所有 X 和 Y 组合

示例应该是包含“X”和“Y”列的 DF

 X      Y
----------
-5     -3
 0     -3
 5     -3
-5      0
 0      0
 5      0
-5      3
 0      3
 5      3

起始条件示例:

a = np.arange(-5,6,5)
b = np.arange(-3,4,3)

我认为如果我先创建 2 个 np.arrays 然后转换为 pandas.df 会快得多。我只是找不到首先创建这些数组的解决方案。

或者有人有更好的主意吗？？？

我已经找到了一个使用 2 个 for 循环将行添加到空 DF 的解决方案。这是极其缓慢和愚蠢的!

Answer 1

您可以使用itertools.product和from_records吗:

from itertools import product

df = pd.DataFrame.from_records([i for i in Product(a,b)])
实际上，您不需要列表理解

df = pd.DataFrame.from_records(product(a,b))
df

输出:

   0  1
0 -5 -3
1 -5  0
2 -5  3
3  0 -3
4  0  0
5  0  3
6  5 -3
7  5  0
8  5  3