11

データ セットの x 値が条件を満たしている場合 (つまり、10 より大きい場合)、xy データのプロット/フィットを実行したいと考えています。

私の試み：

x_values, y_values = loadtxt(fname, unpack=True, usecols=[1, 0])

for x in x_values:
    if x > 10:
        (m,b)=polyfit(x_values,y_values,1)
        yp = polyval([m,b],x_values)
        plot(x_values,yp)
        scatter(x_values,y_values)
    else:
        pass

おそらく、x 値の条件が満たされていない行の xy エントリを削除してから、プロット/フィットする方がよいでしょうか?

4

1 に答える 1

21

もちろん、ブール値のインデックスを使用してください。のようなことができますy = y[x > 10]。

例えば

import numpy as np
import matplotlib.pyplot as plt

#-- Generate some data...-------
x = np.linspace(-10, 50, 100)
y = x**2 + 3*x + 8

# Add a lot of noise to part of the data...
y[x < 10] += np.random.random(sum(x < 10)) * 300

# Now let's extract only the part of the data we're interested in...
x_filt = x[x > 10]
y_filt = y[x > 10]

# And fit a line to only that portion of the data.
model = np.polyfit(x_filt, y_filt, 2)

# And plot things up
fig, axes = plt.subplots(nrows=2, sharex=True)
axes[0].plot(x, y, 'bo')
axes[1].plot(x_filt, y_filt, 'bo')
axes[1].plot(x, np.polyval(model, x), 'r-')

plt.show()

ここに画像の説明を入力

于 2013-02-09T13:16:31.540 に答える