org.simantics.pythonlink/test/org/simantics/pythonlink/test/scripts/NDArray.sts

   1 > import "Python"\r
   2 > ndarray (vector [1, 2, 3])\r
   3 ndarray(3) [1.0, 2.0, 3.0]\r
   4 > ndarrayM 2 2 (vector [1, 2, 3, 4])\r
   5 ndarray(2x2) [[1.0, 2.0], [3.0, 4.0]]\r
   6 > a = ndarrayN (vector [2, 2, 3]) (vector [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12])\r
   7 > a\r
   8 ndarray(2x2x3) [[[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]], [[7.0, 8.0, 9.0], [10.0, 11.0, 12.0]]]\r
   9 > ndarrayDims a\r
  10 vector [2, 2, 3]\r
  11 > ndarrayValues a\r
  12 vector [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0]\r
  13 > ndarrayElement a 5\r
  14 6.0\r
  15 > ndarrayElementN a (vector [0, 1, 2])\r
  16 6.0\r
  17 > b = ndarrayFromList ([[[1, 2, 3], [4, 5, 6]], [[7, 8, 9], [10, 11, 12]]] :: [[[Double]]])\r
  18 > b\r
  19 ndarray(2x2x3) [[[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]], [[7.0, 8.0, 9.0], [10.0, 11.0, 12.0]]]\r
  20 > runPython do\r
  21 >     setPythonNDArrayVariable "foo" (ndarrayM 2 3 (vector [1, 2, 3,  4, 5, 6]))\r
  22 >     executePythonStatement "bar = foo.cumsum(1)"\r
  23 >     getPythonNDArrayVariable "bar"\r
  24 ndarray(2x3) [[1.0, 3.0, 6.0], [4.0, 9.0, 15.0]]\r