gpt4 book ai didi

python - 如何使用 Numpy 重新创建 sigma 符号?

转载 作者:行者123 更新时间:2023-12-04 10:30:46 25 4
gpt4 key购买 nike

我需要使用 Python、numpy 和 matplotlib 绘制以下函数:

See function in image:

对于 N = 5、20 和 60 的值。

我使用以下方法创建了一个奇数列表:

def odd(n):
nums = []
for i in range(1, 2*n, 2):
nums.append(i)
return nums

但我不知道如何在 sigma 函数中使用它,因为我需要改变我的 x 值并对函数求和以求奇数(n)的范围。

最佳答案

这段代码没有循环,应该适用于 x 和 N 的任何值。

x 必须是包含 1 个以上元素的数组或列表

import numpy as np
from numpy import matlib
import matplotlib.pyplot as plt

def psi(x,N):
n=np.arange(0,N,2)+1
sigma = matlib.repmat((-1)**((n-1)/2),len(x),1).T*np.sin(matlib.repmat(n,len(x),1).T*x)
PSI = (2/(N+1))*np.sum(sigma,axis=0)
return PSI
x=np.linspace(0,2*np.pi,50)
N=5
y = psi(x,N)
plt.plot(y)

enter image description here

关于python - 如何使用 Numpy 重新创建 sigma 符号?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/60426653/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com