Основное использование 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() 

Но это не сработало, отслеживающий, как показано ниже:

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