тип numpy.ndarray не определяет метод __round__

Question

это код:

import numpy as np
def f_func(state,time,cd,mass,rho,A):
    """Calculate the differential of state vector as a function of time

        Args:
        state (list): the state vector at time t
        time (float): the time t
        cd (float): the dimensionless drag coefficient
        mass (float): mass of the object in kg
        rho (float): density of air (kg/m3)
        A (float): cross-sectional area of object (kg)

        Returns:
        (list): the differential of the state vector at time t
    """
    # defensive program - check shape of state vector
    assert len(state)==2, "Expected length 2 state vector"
    vy,y = state
    # YOUR CODE HERE

    X = np.array([[vy],[y]])

    # we know d**2 y / d t**2 = a = -g + 1/(2mass)*(cd*rho*A*vy**2)
    d2ydt2 = -g + (1/(2*mass))*(cd*rho*A*vy**2)
    a = d2ydt2
    # WE KNOW d y / d t = vy
    dXdt = np.array([[-g + (1/(2*mass))*(cd*rho*A*(vy)**2)],[vy]])

    return dXdt

Проверено на соответствие:

 from nose.tools import assert_equal, assert_almost_equal
 a,vy = f_func([0.,78.],0.0,0.5,1,1.2,1)
 assert_almost_equal(a, -9.8)
 assert_almost_equal(vy, 0.0)
 a,vy = f_func([-2.,78.],0.0,0.5,1,1.2,1)
 assert_almost_equal(a,-8.6)
 assert_almost_equal(vy,-2) 
 '''

Сообщение об ошибке, которое я не понимаю:

type numpy.ndarray doesn't define __round__ method (error from line 6)

Answer 1

Массивы NumPy не определяют __round__. Кроме того, если массив является 0-мерным, и вы добавляете его к другому числу, он может выглядеть как обычный объект Python, но на самом деле является объектом NumPy.

>>> scalar = np.array(1.0)
>>> type(scalar)
numpy.ndarray
>>> '__round__' in dir(scalar)
False
>>> scalar
array(1.)
>>> scalar + 0
1.0
>>> type(scalar + 0)
numpy.float64

Вы можете использовать функции тестирования NumPy вместо Nose сочень небольшое изменение в вашем коде.

from numpy.testing import assert_equal, assert_almost_equal

Answer 2

Эти переменные не являются числами с плавающей точкой, они являются типами numpy. Numpy-типы не реализуют метод .__round__(), поэтому вы получаете сообщение об ошибке. Эти типы имеют свой собственный метод .round(), который использует numpy. Можете ли вы поменять на:

assert np.isclose(a, -9,8)
assert np.isclose(vy, 0.0)
assert np.isclose(a, -8.6)
assert np.isclose(vy, -2)

Вы можете изменить пределы, чтобы соответствовать поведению assert_almost_equal