从numpy数组中删除虚数可以使用以下方法:
numpy.real()
来获取数组中的实部部分。import numpy as np
arr = np.array([1+2j, 3+4j, 5+6j])
real_arr = np.real(arr)
print(real_arr)
输出:
[1. 3. 5.]
推荐的腾讯云相关产品和产品介绍链接地址:暂无推荐。
numpy.imag()
来获取数组中的虚部部分。import numpy as np
arr = np.array([1+2j, 3+4j, 5+6j])
imag_arr = np.imag(arr)
print(imag_arr)
输出:
[2. 4. 6.]
推荐的腾讯云相关产品和产品介绍链接地址:暂无推荐。
numpy.isreal()
来判断数组中的元素是否为实数。import numpy as np
arr = np.array([1+2j, 3+4j, 5+6j])
is_real = np.isreal(arr)
print(is_real)
输出:
[False False False]
推荐的腾讯云相关产品和产品介绍链接地址:暂无推荐。
numpy.where()
结合上述方法,可以删除数组中的虚数部分。import numpy as np
arr = np.array([1+2j, 3+4j, 5+6j])
real_arr = np.where(np.isreal(arr), arr, np.real(arr))
print(real_arr)
输出:
[1. 3. 5.]
推荐的腾讯云相关产品和产品介绍链接地址:暂无推荐。
需要注意的是,以上方法都是将虚数部分删除或提取出来,返回一个新的数组。原始数组并没有被修改。
领取专属 10元无门槛券
手把手带您无忧上云