如何让数组维度一致?循环绘图出现维度不匹配ValueError报错求助
修复数组维度不匹配问题
问题原因
你的代码通过t、t-1、t-2三个索引计算二阶差分,导致a1/a2/a3的长度比原列表少(160个元素本应生成158个有效计算值,你得到157个说明循环范围少执行了一次),最终和长度160的时间轴维度不匹配。
修复方案
方案1:截断时间轴,匹配y轴长度
直接取时间轴中与a1/a2/a3对应的部分:
- 若
a1长度为157:使用时间数组的[3:]切片(跳过前3个时间点) - 若修正循环后
a1长度为158:使用时间数组的[2:]切片(跳过前2个时间点)
示例代码:
# 假设时间数组名为time_array plt.plot(time_array[2:], a1) # 对应158个y值的情况 # 或者 plt.plot(time_array[3:], a1) # 对应157个y值的情况
方案2:补全y轴数组,匹配时间轴长度
给a1/a2/a3前面补占位值(推荐用NaN,绘图时会自动跳过无效点),让长度变为160:
import numpy as np # 计算需要补的元素个数 pad_count = len(time_array) - len(a1) # 补全数组 a1_padded = [np.nan] * pad_count + a1 a2_padded = [np.nan] * pad_count + a2 a3_padded = [np.nan] * pad_count + a3 # 绘图 plt.plot(time_array, a1_padded) plt.plot(time_array, a2_padded) plt.plot(time_array, a3_padded)
额外优化:修正循环范围与简化计算
- 修正循环范围:如果原列表长度是160,循环应从
t=2到t=159(索引),代码中用range(2, len(x))即可,这样会生成158个有效计算值。 - 简化计算:你的分母
((t + 1) - (t)) ** 2结果恒为1,可以直接省略,代码简化为:
a1.append(x[t] - 2 * x[t - 1] + x[t-2]) a2.append(y[t] - 2 * y[t - 1] + y[t-2]) a3.append(z[t] - 2 * z[t - 1] + z[t-2])
内容的提问来源于stack exchange,提问作者SvylantisCPU
相关产品推荐
相关产品推荐

