Python:連続した値の正負の判定の仕方

Question

リストの中に入っている値の正負を判定し、正負が反転している個所を取得したいです。

a = [ 1, 2, 3, -1, -2, -3, 1, 2, 3]

上記のようなリストがあった場合に、正負が反転している個所を取得するにはどのようにすればよいのでしょうか？上記のリストであれば、三番目（一つ目の-1）と六番目（二つ目の1）になる位置です。

単純に値の正負を比較するのであれば、for文で値を一つずつ取得し、if文で0以上か0以下か判定することができると思うのですが、その反転している位置を知るにはどのようにすればよいのかがわかりません。

x = []
for i, n in enumerate(a):
    if n < 0:
        x.append(i)

上記のように、enumerateでインデックスを取得しながら何らかの処理を行うのかと考えているのですが（全く違ったらすみません…）、条件分岐の仕方がわからないです。
ご教示いただけますと幸いです。よろしくお願いいたします。

Answer 1

こんな方法も。

from itertools import groupby

lst = [1, 2, 3, -1, -2, -3, 1, 2, 3]

index = 0
for k, it in groupby(lst, lambda e: e >= 0):
    print(index)
    index += sum(1 for _ in it)

実行結果 Wandbox

0
3
6

Comments

[Ykkykk]

ご回答いただきありがとうございます。
このような方法でも求めることができるのですね。itertoolsやラムダ式をまだあまり理解できていないので、勉強したいと思います。
ありがとうございます！

Answer 2

numpy の sign() 関数を使うと簡単にかけます。

import numpy as np

a = [ 1, 2, 3, -1, -2, -3, 1, 2, 3]

# 各要素の符号
# signs = np.array(a) >= 0  # 0未満、0以上で区別する場合
signs = np.sign(a)  # 0未満、0、0より大きいかで区別する場合
print(sign)  # [ 1  1  1 -1 -1 -1  1  1  1]

# 隣の要素同士を比較
indices = np.where(signs[:-1] != signs[1:])[0]
print(indices)  # [2 5]

print('符号が変化した箇所: ', indices + 1)  # 符号が変化した箇所:  [3 6]

Comments

[Ykkykk]

ご回答いただきありがとうございます。numpyを使用する方法もあるのですね。
今までnumpyを使用したことがなかったので、これを機に勉強したいと思います。ありがとうございます！

Answer 3

x = []
for i in range(len(a)-1):
    if a[i]*a[i+1] < 0:
       x.append(i)

「かけて負なら符号が替わった」という考え方です。
要素に0が無いことを想定しています。

Comments

[Ykkykk]

ご回答いただきありがとうございます。
「かけて負なら符号が替わった」という風に考えればよいのですね。
言われてみるとそうですが、全く気付かなかったです。目からうろこです。
ありがとうございます！