python エラーへの対処

Question

エラーが出ていますが，どこへのエラーなのか見当がつきません・・・．

アドバイスいただけませんか？

エラーは

ValueError: operands could not be broadcast together with shapes (16386,) (16384,)

です．

2つのファイルは数値が各行にあり，数値の個数は同じです

import matplotlib.pyplot as plt
import numpy as np

x = [float(s) for s in open('./test1.txt')]
y = [float(s) for s in open('./test2.txt')]


from swan import pycwt

Fs = 1/0.01
omega0 = 6

freqs=np.arange(0.1,20,0.025)
r=pycwt.cwt_f(y,freqs,Fs,pycwt.Morlet(omega0))
rr=np.abs(r)

plt.rcParams['figure.figsize'] = (20,6)
fig = plt.figure()
ax1 = fig.add_axes([0.1,0.75,0.7,0.2])
ax2 = fig.add_axes([0.1,0.1,0.7,0.60],sharex=ax1)
ax3 = fig.add_axes([0.83,0.1,0.03,0.6])

ax1.plot(x,y,'k')

img = ax2.imshow(np.flipud(rr),extent=[0,20,freqs[0],freqs[-1]],
                 aspect='auto',interpolation='nearest')
twin_ax = ax2
#logスケールにする
#twin_ax.set_yscale('log')
twin_ax.set_xlim(0, 20)
twin_ax.set_ylim(0.1, 20)
ax2.tick_params(which='both', labelleft=False, left=False)
twin_ax.tick_params(which='both', labelleft=True, left=True, labelright=False)

fig.colorbar(img,cax=ax3)

plt.show ()

Comments

[ppaul]

print(len(x))
print(len(y))
の実行結果を質問に追加してください。

[shisham]

1901
1901

と出てきました．データの個数と一致しています

[shisham]

y = np.array([float(s) for s in open('./test2.txt')])

nemoplusさんのおっしゃっていたように変えてみましたが，エラーは同じように出ています

Answer 1

pycwt.cwt_f 関数への第１引数が問題のようです。

y = [float(s) for s in open('./test2.txt')]
...
r=pycwt.cwt_f(y,freqs,Fs,pycwt.Morlet(omega0))

test2.txt の中身が不明ですが、numpy array に変換するといかがでしょうか。

y = np.array([float(s) for s in open('./test2.txt')])

訂正
numpy array への変換は不要でした。
y は 10件以上のリストであれば動作するようです。
http://cell.biophys.msu.ru/static/swan/howtos/simple-usage.html