Python pmap1の例

コード例 #1

ファイル: test_windowed.py プロジェクト: Abramovuch/parakeet

def test_window_shapes():
  x = np.array([0,1,2,3,4])
  def winshape(x):
    return x.shape[0] 
  y = parakeet.pmap1(winshape, x, 3)
  
  expected = np.array([2,3,3,3,2])
  assert y.shape == expected.shape, "Expected shape %s but got %s" % (expected.shape, y.shape)
  assert np.all(y == expected), "Expected array %s but got %s" % (expected, y)

コード例 #2

def test_window_shapes():
    x = np.array([0, 1, 2, 3, 4])

    def winshape(x):
        return x.shape[0]

    y = parakeet.pmap1(winshape, x, 3)

    expected = np.array([2, 3, 3, 3, 2])
    assert y.shape == expected.shape, "Expected shape %s but got %s" % (
        expected.shape, y.shape)
    assert np.all(
        y == expected), "Expected array %s but got %s" % (expected, y)

コード例 #3

def winmap_avg1d(x, w=3):
    return parakeet.pmap1(avg1d, x, w)

コード例 #4

ファイル: test_windowed.py プロジェクト: Tillsten/parakeet

def winmap_avg1d(x, w=3):
    return parakeet.pmap1(avg1d, x, w)