如何计算数组中每个值的公式？

Question

我试图了解如何使用 numpy 计算不同时间的公式。编写代码的方式给出了 y 大于 0 的所有值。我正在试验如何获取所有 y 的值。

有没有人可以解释我的部分：ft = t * [y >= 0.0 ]。如何使用括号内的部分？

from numpy import *

g = 10.0
h0 = 10.0

t = arange(0, 10.1 ,0.1)
y = h0 - 0.5*g*t*t

ft = t * [y >= 0.0 ]


print(ft)

这是输出，但我想查看计算的所有值。所以我做了一些实验，但我无法弄清楚如何去做以及如何; [y >= 0.0] 部分完全有效。

[[0.  0.1 0.2 0.3 0.4 0.5 0.6 0.7 0.8 0.9 1.  1.1 1.2 1.3 1.4 0.  0.  0.
  0.  0.  0.  0.  0.  0.  0.  0.  0.  0.  0.  0.  0.  0.  0.  0.  0.  0.
  0.  0.  0.  0.  0.  0.  0.  0.  0.  0.  0.  0.  0.  0.  0.  0.  0.  0.
  0.  0.  0.  0.  0.  0.  0.  0.  0.  0.  0.  0.  0.  0.  0.  0.  0.  0.
  0.  0.  0.  0.  0.  0.  0.  0.  0.  0.  0.  0.  0.  0.  0.  0.  0.  0.
  0.  0.  0.  0.  0.  0.  0.  0.  0.  0.  0. ]]

如果我使用 [y] 而不是 [y >= 0.0] 我会得到以下结果：

[[ 0.000000e+00  9.950000e-01  1.960000e+00  2.865000e+00  3.680000e+00
   4.375000e+00  4.920000e+00  5.285000e+00  5.440000e+00  5.355000e+00
   5.000000e+00  4.345000e+00  3.360000e+00  2.015000e+00  2.800000e-01
  -1.875000e+00 -4.480000e+00 -7.565000e+00 -1.116000e+01 -1.529500e+01
  -2.000000e+01 -2.530500e+01 -3.124000e+01 -3.783500e+01 -4.512000e+01
  -5.312500e+01 -6.188000e+01 -7.141500e+01 -8.176000e+01 -9.294500e+01
  -1.050000e+02 -1.179550e+02 -1.318400e+02 -1.466850e+02 -1.625200e+02
  -1.793750e+02 -1.972800e+02 -2.162650e+02 -2.363600e+02 -2.575950e+02
  -2.800000e+02 -3.036050e+02 -3.284400e+02 -3.545350e+02 -3.819200e+02
  -4.106250e+02 -4.406800e+02 -4.721150e+02 -5.049600e+02 -5.392450e+02
  -5.750000e+02 -6.122550e+02 -6.510400e+02 -6.913850e+02 -7.333200e+02
  -7.768750e+02 -8.220800e+02 -8.689650e+02 -9.175600e+02 -9.678950e+02
  -1.020000e+03 -1.073905e+03 -1.129640e+03 -1.187235e+03 -1.246720e+03
  -1.308125e+03 -1.371480e+03 -1.436815e+03 -1.504160e+03 -1.573545e+03
  -1.645000e+03 -1.718555e+03 -1.794240e+03 -1.872085e+03 -1.952120e+03
  -2.034375e+03 -2.118880e+03 -2.205665e+03 -2.294760e+03 -2.386195e+03
  -2.480000e+03 -2.576205e+03 -2.674840e+03 -2.775935e+03 -2.879520e+03
  -2.985625e+03 -3.094280e+03 -3.205515e+03 -3.319360e+03 -3.435845e+03
  -3.555000e+03 -3.676855e+03 -3.801440e+03 -3.928785e+03 -4.058920e+03
  -4.191875e+03 -4.327680e+03 -4.466365e+03 -4.607960e+03 -4.752495e+03
  -4.900000e+03]]

我想知道如何使用 numpy 一次计算不同时间间隔的公式的所有结果。

谢谢，

Answer 1

y >= 0.0 为您提供一个布尔数组，其中包含 True/False，具体取决于条件 y >= 0.0 的满足情况。当您将它作为[y >= 0.0] 包含在[] 中时，您会得到一个包含单个布尔数组的列表，正如@nicola 在下面的cmets 中指出的那样。

[array([ True,  True,  True,  True, True, False, False, False,...
   ... False, False, False, False])]

现在您将其与您的 arange 数组相乘，当 * 运算符的右侧为 False 时，您将得到 0，并在右侧为 arange 时为您提供实际值* 运营商是True

Answer 2

数组 [y >= 0.0] 产生布尔值数组。即如果 y>=0 则为 1，否则为 0。然后将 1 和 0 的数组乘以 t。

但是从你的问题中我不清楚你想用它做什么。