配列の要素を３個ずつ合計して計算したい

Question

配列の要素を３個ずつ合計して計算したいです。

# coding: utf-8

first_lines = input()
s = first_lines.rstrip().split()
a_str = list(map(int,s))

num_array =[]

for i in range(a_str[0]):
  second_lines = input()
  t = second_lines.rstrip().split()
  b_str = list(map(int,t))
  num_array.extend(b_str)

for_num =[]
for i in range(a_str[0]*2):
    for_num.append(i*3)
x=0
three_num =[]

for j in range(a_str[0]*2):
    k = j*3
    l = k+3  
    three_num.append(num_array[k:l])

sum_num=[]

for i in range(a_str[0]*2):
    x =three_num[i]
    y = sum(x)
    sum_num.append(y)

print(sum_num)

sum_num2=[]
sum_num3=[]

deci =  int(a_str[0]/a_str[1])

for j in range(a_str[0]*2):
    if j % 2 == 0:
      sum_num2.append(sum_num[j])

    else:
      sum_num3.append(sum_num[j])

print(sum_num2)
print(sum_num3)

とコードを書きました。

print(sum_num2)
print(sum_num3)
で

[6, 6, 6, 15, 15, 15] 
[15, 15, 15, 6, 6, 6]

と出力されました。
これらの配列を、配列ごとに配列の要素を３個ずつ合計して計算したいです。
例えば、
6, 6, 6 を計算して18、
15, 15, 15 を計算して 45、

15, 15, 15 を計算して 45、
6, 6, 6 を計算して18、
なので、最終的には合計した値を配列ごとに入力して
[18,45]
[45,18]
と出力させたいです。
だけど、for文もしくは配列のスライスを使って要素を３個ずつ取り出す方法がわかりません。
どのように書けますか？

Answer 1

こんな感じでしょうか？

sum_num = [6, 6, 6, 15, 15, 15]
print([sum(sum_num[i:i+3]) for i in range(0,len(sum_num),3)])
#=> [18, 45]

Comments

[退会済みユーザー]

ありがとうございます！すごくすっきり書けますね！ひとつ疑問なのですが、print文の中でfor文を回すことはできるのでしょうか？print文の中で、出力させたいものを[] で囲っている点がよくわからなくて..。この[]の事は何と呼ぶのでしょうか？

[magichan]

'[]' は通常のリストを表す'[]' です
for文があるのはリストの内包表記となります。

ですので、
「print文内でfor文を回すことができる」
というよりは
「内包表記によりfor文を使ってリストを作成している→その結果をprintしている」
と理解した方が正しいお思います。

[退会済みユーザー]

なるほど！ありがとうございます！

Answer 2

python3

iarr = [6, 6, 6, 15, 15, 15]

stotal = 0 # 小計
rcount = 0 # ループカウンタ
for v in iarr:
    stotal += v # 小計に加算
    rcount += 1 # カウントをすすめる
    if rcount % 3 == 0: # 3回ごとに実行
        print(stotal)   # 合計を取得
        stotal = 0      # 小計をリセット

Comments

[intelf___]

回答はヒントになります。細かい部分は自分で実装してください。
print(stotal)を表示せず配列に格納して返す関数を作れば複数の配列でも利用できそうですね。

Answer 3

# 自分の得意な言語で
# Let's チャレンジ！！

自分の得意な言語で、とあるので、好きな言語で答えてよろしいのでしょうか？

Comments

[退会済みユーザー]

すみません、その部分はないものと考えてください。Pythonでお願いします