-1

我正在从文本文件中读取数据，但是当我这样做时，我需要 在绘图函数中将这个值倍增，例如3*sqrt(col1)= 。x1.append(3*math.sqrt(float(p[1])))如何在绘图之前使用多个列号数据？例如，我将通过 3*sqrt(col3) 和绘制该数据之后的多个 col3 数据。

#-------input.dat---------
#   x        y     z
# col 1    col 2  col 3
# 3          5      5
# 5          6      4
# 7          7      3
import matplotlib.pyplot as plt
import numpy as np
import pylab as pl
import math

data = open('input.dat')
lines = data.readlines()
data.close()
x1=[]
y1=[]
z1=[]
plt.plot(1)
for line in lines[2:]:
p= line.split()
x1.append(3*math.sqrt(float(p[1])))
y1.append(3*math.sqrt(float(p[2])))
z1.append(3*math.sqrt(float(p[3])))
x=np.array(x1)
y=np.array(y1)
z=np.array(z1)
plt.subplot(311)
plt.plot(x,'b',label=" X figure ")
plt.subplot(312)
plt.plot(y,'r',label=" Y figure ")
plt.subplot(313)
plt.plot(x,z,'g',label=" X,Z figure ")
plt.show()
4

1 回答 1

2

同样，如果您numpy从一开始就使用数组，这会更容易。

通过读取我在上一个问题中向您展示的数据，您的数据将已经在numpy数组中。然后，您可以使用该numpy.sqrt函数对数组按元素执行平方根运算。

#-------input.dat---------
#   x        y     z
# col 1    col 2  col 3
# 3          5      5
# 5          6      4
# 7          7      3
import matplotlib.pyplot as plt
import numpy as np

data = np.genfromtxt('input.dat', skip_header=2)

x = 3. * np.sqrt(data[:, 0])
y = 3. * np.sqrt(data[:, 1])
z = 3. * np.sqrt(data[:, 2])

plt.subplot(311)
plt.plot(x, 'b', label=" X figure ")
plt.subplot(312)
plt.plot(y, 'r', label=" Y figure ")
plt.subplot(313)
plt.plot(x, z, 'g', label=" X,Z figure ")
plt.show()

在此处输入图像描述

但是，如果您真的想坚持使用旧代码，可以通过

  1. 固定压痕，

  2. 将索引更改为p[0], p[1]and p[2]（而不是p[1], p[2]and p[3]）

此代码生成与上面相同的图：

import matplotlib.pyplot as plt
import numpy as np
import pylab as pl
import math

data = open('input.dat')
lines = data.readlines()
data.close()
x1=[]
y1=[]
z1=[]
plt.plot(1)
for line in lines[2:]:
    p= line.split()
    x1.append(3*math.sqrt(float(p[0])))
    y1.append(3*math.sqrt(float(p[1])))
    z1.append(3*math.sqrt(float(p[2])))
x=np.array(x1)
y=np.array(y1)
z=np.array(z1)
plt.subplot(311)
plt.plot(x,'b',label=" X figure ")
plt.subplot(312)
plt.plot(y,'r',label=" Y figure ")
plt.subplot(313)
plt.plot(x,z,'g',label=" X,Z figure ")
plt.show()
于 2018-06-15T10:09:04.133 回答