从数组中选择数据

时间:2014-06-10 03:05:34

标签: python loops

我有数组数据,(T)。 T是温度随时间(t)和高度(h)的函数。我想在特定高度中选择T的值,例如,在100 km高度中,然后在2D图形中绘制它。 为什么这个循环不起作用?

for j in range(len(tim1)):
    for i in range(len(height1)):
        if height1 == 350. :
            print(i,j,T)

错误:

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "C:\Python27\lib\site-packages\spyderlib\widgets\externalshell\sitecustomize.py", line 540, in runfile
    execfile(filename, namespace)

    if height1 == 350. :
ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

有人可以帮助我吗?提前谢谢

1 个答案:

答案 0 :(得分:1)

我可能误解了这个问题,但您可能只需确保在测试条件时索引列表:

for j in range(len(tim1)):
    for i in range(len(height1)):
        if height1[i] == 350. :
            print(i,j,T)