pythonのnp.arrayがよくわからない

Question

Atcoder Educational DP Contest B-Frog2について、この回答で勉強しようと思ったのですがわからない部分があります。
heights[i] - heights[start:i]　の部分が、配列同士の引き算で要素数も違うのになぜできてしかも結果が数値になっているのかよくわからず、とりあえず　print(heights[i] - heights[start:i])　として確認してみようとしたのですが、この一文をいれると大量のエラーが出てしまいます。（この一文を入れなければ正しく動きます。）どういうことなのでしょうか。

import math
import sys
import os
import numpy as np
f = open('input.txt', 'r')
sys.stdin = f

N, K = map(int, input().split())
H = [int(i) for i in list(input().split())]
dp = np.zeros(N, dtype=int)
heights = np.array(H)
for i in range(1, N):
    start = max(0, i - K)
    print(heights[i] - heights[start:i])
    dp[i] = (dp[start:i] + np.abs(heights[i] - heights[start:i])).min()
print(dp)

Comments

[_Victorique__]

エラー詳細を載せてください

Answer 1

エラーメッセージをみたいですね。
コードをみると python3系だとうまく行けて、
python2系だと

input().split()

を

raw_input().split()

と置き換える必要があります。

こちらでは実際に動いています。

Comments

[Sonono]

みなさん、回答ありがとうございました。あの後アプリがクラッシュしてしまったので放置していて今もう一度やってみたところ、エラーはでませんでした...。原因はわかりませんが、お騒がせしました。疑問に思っていた箇所もnp.absについてわかっていなかったからということがわかりましたので解決いたしました。ありがとうございました。