Python：使用matplotlib绘制x-y图

Question

我想画一些数据图。第一列是x轴的数据。但是matplotlib没有画出来。这是哪里出错了呢？

import numpy as np
from numpy import cos
from scipy import *
from pylab import plot, show, ylim, yticks
from matplotlib import *
from pprint import pprint

n1 = 1.0
n2 = 1.5

#alpha, beta, intensity
data = [
    [10,    22,     4.3],
    [20,    42,     4.2],
    [30,    62,     3.6],
    [40,    83,     1.3],
    [45,    102,    2.8],
    [50,    123,    3.0],
    [60,    143,    3.2],
    [70,    163,    3.8],
    ]

for i in range(len(data)):
    rhotang1 = (n1 * cos(data[i][0]) - n2 * cos(data[i][1]))
    rhotang2 = (n1 * cos(data[i][0]) + n2 * cos(data[i][1]))
    rhotang = rhotang1 / rhotang2
    data[i].append(rhotang) #append 4th value

pprint(data)
x = data[:][0]
y1 = data[:][2]
y3 = data[:][3]
plot(x, y1, x, y3)
show()

补充说明：http://paste.pocoo.org/show/205534/ 但是它还是不管用。

Answer 1

试试这个：

#fresnel formula

import numpy as np
from numpy import cos
from scipy import *
from pylab import plot, show, ylim, yticks
from matplotlib import *
from pprint import pprint

n1 = 1.0
n2 = 1.5

#alpha, beta, intensity
data = np.array([
    [10,    22,     4.3],
    [20,    42,     4.2],
    [30,    62,     3.6],
    [40,    83,     1.3],
    [45,    102,    2.8],
    [50,    123,    3.0],
    [60,    143,    3.2],
    [70,    163,    3.8],
    ])

# Populate arrays
x = np.array([row[0] for row in data])
y1 = np.array([row[1] for row in data])
rhotang1 = n1*cos(data[:,0]) - n2*cos(data[:,1])
rhotang2 = n1*cos(data[:,0]) + n2*cos(data[:,1])
y3 = rhotang1 / rhotang2

plot(x, y1, 'r--', x, y3, 'g--')
show()

Answer 2

你可以通过把数据转换成一个numpy数组来实现这个。

data = np.array(data) # insert this new line after your appends

pprint(data)
x = data[:,0]    # use the multidimensional slicing notation
y1 = data[:,2]
y3 = data[:,3]
plot(x, y1, x, y3)

还有几点补充：

你可以用numpy以更清晰和更高效的方式来进行计算，像这样：

data = np.array(data)
rhotang1 = n1*cos(data[:,0]) - n2*cos(data[:,1])
rhotang2 = n1*cos(data[:,0]) + n2*cos(data[:,1])
y3 = rhotang1 / rhotang2

按照你现在的写法，你的计算可能得不到你想要的结果，因为cos等函数是以弧度为输入的，而你的数字看起来像是度数。

Answer 3

x = data[:][0]
y1 = data[:][2]
y3 = data[:][3]

这些代码行并不是你想的那样。

首先，它们从数组中取出了一部分，但实际上是整个数组的一个副本（也就是说，还是原来的内容），然后它们从这个数组中提取了第0行、第2行或第3行，而不是列。

你可以试试

x = [row[0] for row in x]

等等。