如何在for循环中构建包含x、y、z值的3列矩阵？

API 探路者

2026-8-12

解决方案

问题分析

你的代码目标是生成每行包含(x,y,z)的N×3矩阵（N为time的长度），但存在几个明显问题：

未定义变量r和h，运行会直接报错
用np.vstack循环拼接数组效率极低，尤其当time长度较大时
可以利用numpy的向量化运算替代循环，代码更简洁高效

改进方案

1. 补全缺失变量

首先必须定义r和h的具体值，示例如下：

r = 10  # 根据你的实际需求修改
h = 2   # 根据你的实际需求修改

2. 向量化操作实现（推荐）

numpy支持对整个数组批量运算，无需逐元素循环，速度和代码简洁度都更优：

import numpy as np

time = np.arange(0, 100, 1)
fphi = 2  # kampo phi daznis
ftheta = 3  # kampo i daznis
Amp = np.pi/2
r = 10  # 补全r的取值
h = 2   # 补全h的取值

# 批量计算phi和theta
phi = 2 * np.pi * fphi * time
theta = Amp * np.sin(2 * np.pi * ftheta * time)

# 批量生成x、y、z数组
x = r * np.cos(phi)
y = r * np.sin(phi) * np.cos(theta) - h * np.sin(theta)
z = r * np.sin(phi) * np.sin(theta) + h * np.cos(theta)

# 合并为N×3的目标矩阵
points = np.column_stack((x, y, z))

# 后续转置直接调用.T属性即可
transposed_points = points.T

3. 循环实现（不推荐，仅做参考）

如果一定要用循环，建议先预分配数组内存，避免每次vstack的内存重新分配开销：

import numpy as np

time = np.arange(0, 100, 1)
fphi = 2  # kampo phi daznis
ftheta = 3  # kampo i daznis
Amp = np.pi/2
r = 10
h = 2

# 预分配N×3的空数组，N为time的长度
points = np.empty((len(time), 3))

for idx, p in enumerate(time):
    phi = 2 * np.pi * fphi * p
    theta = Amp * np.sin(2 * np.pi * ftheta * p)
    x = r * np.cos(phi)
    y = r * np.sin(phi) * np.cos(theta) - h * np.sin(theta)
    z = r * np.sin(phi) * np.sin(theta) + h * np.cos(theta)
    points[idx] = [x, y, z]

# 执行转置
transposed_points = points.T