pythonの配列基本的な使い方：step-function

Question

私はPythonの初心者です。配列やリストの使い方について混乱します。

：これは以下のように、トレースバックが動作しませんでした

# -*- coding: utf-8 -* 
import numpy as np 
import matplotlib.pyplot as plt 
import matplotlib as mpl 
import math 
from pylab import * 

i = np.arange(2,5,0.1) 
t = 1+i 
Light = 10 
if t > 3 : 
    Light = 5 

plt.figure('God Bless: Lightcure') 
plt.plot(i,Light) 
plt.show() 

しかし：私はちょうど2つのセクションにデータを分離するようにしたいが、私はどのように知りませんが、以下のようにではなく、基本的な使用方法で私を助けてください

Traceback (most recent call last): 
File "1.py", line 11, in <module> 
if t> 3 : 
ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all() 

ありがとうございました！

Answer 1

# True is where condition is satisfied: numpy.ndarray([False, False, ..., True, True]) 
mask = t > 3 

# Uninitialized array with same shape as t 
Light = numpy.empty_like(t) 

# Light elements set to 5 where corresponding mask elements are True 
Light[mask] = 5 

# Light elements set to 10 where corresponding mask elements are False 
Light[~mask] = 10