リストに格納されている0を計算せずに他の値は計算したい。

Question

前提

以下のような二次元リストを作成しました。

x = [[-2, 7, -2], [0, -4, -4], [2, -3, -2], [-3, -4, -4], [-4, 0, -4]]

実現したいこと

これらの数字に+1してその後に3で割り算したいです。しかし、x[1][0]とx[4][2]は0なので計算せず0は0のままにしたいです。
つまり、このような計算手順になります。

①-2を足したとき、

x_1 = [[-4, 5, -4],[0, -6, -6],[0, -5, -4],[-5, -6, -6],[-6, 0, -6]]

②3で割った時、

x_2 = [[-1.33, 1.67, -1.33], [0, -2, -2], [0, -1.67, -1.33], [-1.67, -2, -2], [-2, 0, -2]]

となります。

試したこと

まず、①の計算をする時に、数字が0であった場合は計算しないという風なコードを作ればいいと思ったのでif文を使いたくなりました。

x = [[-2, 7, -2], [0, -4, -4], [2, -3, -2], [-3, -4, -4], [-4, 0, -4]]

for s in x:
    for x_1 in s:
        print(x_1)

このようなコードを書くと、リストxに格納されているすべての数字が取り出せます。
続いて、print(x_1 - 2)としたときは、このようになります。

x = [[-2, 7, -2], [0, -4, -4], [2, -3, -2], [-3, -4, -4], [-4, 0, -4]]

for s in x:
    for x_1 in s:
        print(x_1  -2)
#出力結果
-4
5
-4
-2
-6
-6
0
-5
-4
-5
-6
-6
-6
-2
-6

すべての数字に-2を足した値が出力されました。
続いて、if文でリストxの中身が0だった場合計算しないというコードを作りました。

x = [[-2, 7, -2], [0, -4, -4], [2, -3, -2], [-3, -4, -4], [-4, 0, -4]]

for s in x:
    for x_1 in s:
        if x_1 == 0:
            continue
        else:
            print(x_1  -2)

#出力結果
-4
5
-4
-6
-6
0
-5
-4
-5
-6
-6
-6
-6

これだと、x_1にはリストxにあった0が格納されていない(スキップされた)ので実現したいことと異なってしまいました。
ここまで、考えてどのようにすれば解決できるのか分からなくなってしまったので質問します。
ここが分かれば3で割った時はprint(x_1 -2)をprint(x_2/3)にすれば値が求められると思います。

Comments

[meg_]

> これだと、x_1にはリストxにあった0が格納されていない(スキップされた)
「x_1には」「リストxにあった0が格納」されているが「スキップ」しているだけですよね？

[hoshi-takanori]

単純に、x_1 が 0 の時にスキップせずに (continue の代わりに)、x_1 をそのまま表示すれば良いような。

Answer 1

Numpy を使う場合。

import numpy as np
from pprint import pprint

x = [[-2, 7, -2], [0, -4, -4], [2, -3, -2], [-3, -4, -4], [-4, 0, -4]]

nx = np.array(x)
nx[nx != 0] -= 2
y = (nx / 3.0).round(2).tolist()
pprint(y)

#
[[-1.33, 1.67, -1.33],
 [0.0, -2.0, -2.0],
 [0.0, -1.67, -1.33],
 [-1.67, -2.0, -2.0],
 [-2.0, 0.0, -2.0]]

Comments

[退会済みユーザー]

ありがとうございます！

Answer 2

単純に、 x_1 == 0 のときも print(x1) を入れたらよいかなと思います。
もしくは、

for s in x:
    for x_1 in s:
        print(x1 - 2 if x1 != 0 else 0)

でもよいかなと思います。

ただ、Numpy使えるなら、この回答 がよいかなと思います。

Comments

[退会済みユーザー]

continueにしてるからダメだったんですね、、気づきませんでした。ありがとうございます！

Answer 3

これらの数字に+1してその後に3で割り算したいです。しかし、x[1][0]とx[4][2]は0なので計算せず0は0のままにしたいです。

+1 ではなく -2 ですね。x[4][2] ではなく x[4][1] が 0 ですね。

x = [[-2, 7, -2], [0, -4, -4], [2, -3, -2], [-3, -4, -4], [-4, 0, -4]]

x1 = [[y-2 if y else 0 for y in s] for s in x]
print(x1)

x2 = [[round(y/3, 2) for y in s] for s in x1]
print(x2)

小数点以下が 0 の場合、.0 を表示しないようにしたいですか？

Answer 4

if文で０のときはprintさえスキップするようになってますね

Comments

[退会済みユーザー]

continueのところをprintに変更したらできました。ありがとうございます！