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

在matplotlib绘图中使用cos函数

2 个答案: