使用 cos 函数使用 matplotlib 进行绘图

Question

我有一个简单的文本文件，由 8 列组成，我使用 loadtxt 函数读取它。我想绘制为 y 轴列 2-列 5 和 x 轴列 1-列 4 除以 cos（列 2-列 5），所以我把这个命令

>>> 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] ] ) )