python - 在 matplotlib 绘图中使用 cos 函数

Question

我有一个由 8 列组成的简单文本文件，我使用 loadtxt 函数读取它。我想绘制为 y 轴 column2-column5 和 x 轴 column1-column4 除以 cos(column2-column5) 所以我把这个命令

>>> y = data[:,2] - data[:,5]
>>> x = (data[:,1] - data[:,4])/cos(y)
and it gave this error

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
NameError: name 'cos' is not defined

问题是什么？

Answer 1

您可能想使用 numpy 数组。然后你可以使用元素明智的划分。此外，numpy 提供了所有必要的数学函数。

import numpy as np
d = np.asarray(data)
y = d[:,2] - d[:,5]
x = (d[:,1] - d[:,4])/np.cos(y)

Answer 2

cos在math模块中（以及其他数学库numpy等）

import math
math.cos(3) 

为了numpy

import numpy
numpy.cos( np.array( [ [1,2], [3,4] ] ) )