{ "cells": [ { "cell_type": "markdown", "metadata": {}, "source": [ "This is Ruby/Numo::NArray version of\n", "[100 numpy exercises](http://www.labri.fr/perso/nrougier/teaching/numpy.100/)\n", "([Repository](https://github.com/rougier/numpy-100))" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 1. Import the numpy package under the name `np` (★☆☆)\n", "\n", "\n", "Python:\n", "```python\n", "import numpy as np\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 1, "metadata": { "collapsed": false }, "outputs": [ { "data": { "text/plain": [ "true" ] }, "execution_count": 1, "metadata": {}, "output_type": "execute_result" } ], "source": [ "require \"numo/narray\"" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 2. Print the numpy version and the configuration (★☆☆)\n", "\n", "\n", "Python:\n", "```python\n", "print(np.__version__)\n", "np.show_config()\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 2, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "\"0.9.0.3\"\n" ] }, { "data": { "text/plain": [ "\"0.9.0.3\"" ] }, "execution_count": 2, "metadata": {}, "output_type": "execute_result" } ], "source": [ "p Numo::NArray::VERSION" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 3. Create a null vector of size 10 (★☆☆)\n", "\n", "\n", "Python:\n", "```python\n", "Z = np.zeros(10)\n", "print(Z)\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 3, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Numo::DFloat#shape=[10]\n", "[0, 0, 0, 0, 0, 0, 0, 0, 0, 0]\n" ] }, { "data": { "text/plain": [ "Numo::DFloat#shape=[10]\n", "[0, 0, 0, 0, 0, 0, 0, 0, 0, 0]" ] }, "execution_count": 3, "metadata": {}, "output_type": "execute_result" } ], "source": [ "z = Numo::DFloat.zeros(10)\n", "p z" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 4. How to find the memory size of any array (★☆☆)\n", "\n", "\n", "Python:\n", "```python\n", "Z = np.zeros((10,10))\n", "print(\"%d bytes\" % (Z.size * Z.itemsize))\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 4, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "800 bytes" ] } ], "source": [ "z = Numo::DFloat.zeros(10,10)\n", "printf \"%d bytes\", z.byte_size" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 5. How to get the documentation of the numpy add function from the command line? (★☆☆)\n", "\n", "Python:\n", "\n", "$ `python -c \"import numpy; numpy.info(numpy.add)\"`\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 5, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "\u001b[0m\u001b[1;32mNumo::DFloat#+\u001b[m\n", "\n", "(from gem numo-narray-0.9.0.3)\n", "\u001b[32mImplementation from DFloat\u001b[m\n", "------------------------------------------------------------------------------\n", " +(p1)\n", "\n", "------------------------------------------------------------------------------\n", "\n", "Binary add. @overload + other @param [Numo::NArray,Numeric] other @return\n", "[Numo::NArray] self + other\n", "\n", "\n" ] } ], "source": [ "ri 'Numo::DFloat#+'" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 6. Create a null vector of size 10 but the fifth value which is 1 (★☆☆)\n", "\n", "\n", "Python:\n", "```python\n", "Z = np.zeros(10)\n", "Z[4] = 1\n", "print(Z)\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 6, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Numo::DFloat#shape=[10]\n", "[0, 0, 0, 0, 1, 0, 0, 0, 0, 0]\n" ] }, { "data": { "text/plain": [ "Numo::DFloat#shape=[10]\n", "[0, 0, 0, 0, 1, 0, 0, 0, 0, 0]" ] }, "execution_count": 6, "metadata": {}, "output_type": "execute_result" } ], "source": [ "z = Numo::DFloat.zeros(10)\n", "z[4] = 1\n", "p z" ] }, { "cell_type": "markdown", "metadata": { "collapsed": true }, "source": [ "#### 7. Create a vector with values ranging from 10 to 49 (★☆☆)\n", "\n", "\n", "Python:\n", "```python\n", "Z = np.arange(10,50)\n", "print(Z)\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 7, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Numo::DFloat#shape=[40]\n", "[10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, ...]\n" ] }, { "data": { "text/plain": [ "Numo::DFloat#shape=[40]\n", "[10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, ...]" ] }, "execution_count": 7, "metadata": {}, "output_type": "execute_result" } ], "source": [ "z = Numo::DFloat[10..49]\n", "p z" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 8. Reverse a vector (first element becomes last) (★☆☆)\n", "\n", "\n", "Python:\n", "```python\n", "Z = np.arange(50)\n", "Z = Z[::-1]\n", "print(Z)\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 9, "metadata": { "collapsed": false }, "outputs": [ { "data": { "text/plain": [ "Numo::Int32(view)#shape=[50]\n", "[49, 48, 47, 46, 45, 44, 43, 42, 41, 40, 39, 38, 37, 36, 35, 34, 33, 32, ...]" ] }, "execution_count": 9, "metadata": {}, "output_type": "execute_result" } ], "source": [ "z = Numo::Int32.new(50).seq\n", "z = z.reverse" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 9. Create a 3x3 matrix with values ranging from 0 to 8 (★☆☆)\n", "\n", "\n", "Python:\n", "```python\n", "Z = np.arange(9).reshape(3,3)\n", "print(Z)\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 10, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Numo::Int32#shape=[3,3]\n", "[[0, 1, 2], \n", " [3, 4, 5], \n", " [6, 7, 8]]\n" ] }, { "data": { "text/plain": [ "Numo::Int32#shape=[3,3]\n", "[[0, 1, 2], \n", " [3, 4, 5], \n", " [6, 7, 8]]" ] }, "execution_count": 10, "metadata": {}, "output_type": "execute_result" } ], "source": [ "z = Numo::Int32.new(3,3).seq\n", "p z" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 10. Find indices of non-zero elements from \\[1,2,0,0,4,0\\] (★☆☆)\n", "\n", "\n", "Python:\n", "```python\n", "nz = np.nonzero([1,2,0,0,4,0])\n", "print(nz)\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 11, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Numo::Int32#shape=[3]\n", "[0, 1, 4]\n" ] }, { "data": { "text/plain": [ "Numo::Int32#shape=[3]\n", "[0, 1, 4]" ] }, "execution_count": 11, "metadata": {}, "output_type": "execute_result" } ], "source": [ "nz = Numo::NArray[1,2,0,0,4,0].ne(0).where\n", "p nz" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 11. Create a 3x3 identity matrix (★☆☆)\n", "\n", "\n", "Python:\n", "```python\n", "Z = np.eye(3)\n", "print(Z)\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 12, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Numo::DFloat#shape=[3,3]\n", "[[1, 0, 0], \n", " [0, 1, 0], \n", " [0, 0, 1]]\n" ] }, { "data": { "text/plain": [ "Numo::DFloat#shape=[3,3]\n", "[[1, 0, 0], \n", " [0, 1, 0], \n", " [0, 0, 1]]" ] }, "execution_count": 12, "metadata": {}, "output_type": "execute_result" } ], "source": [ "z = Numo::DFloat.eye(3)\n", "p z" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 12. Create a 3x3x3 array with random values (★☆☆)\n", "\n", "\n", "Python:\n", "```python\n", "Z = np.random.random((3,3,3))\n", "print(Z)\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 13, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Numo::DFloat#shape=[3,3,3]\n", "[[[0.0617545, 0.373067, 0.794815], \n", " [0.201042, 0.116041, 0.344032], \n", " [0.539948, 0.737815, 0.165089]], \n", " [[0.0508827, 0.108065, 0.0687079], \n", " [0.904121, 0.478644, 0.342969], \n", " [0.164541, 0.74603, 0.138994]], \n", " [[0.411576, 0.292532, 0.869421], \n", " [0.0854984, 0.688965, 0.159977], \n", " [0.279215, 0.625155, 0.676329]]]\n" ] }, { "data": { "text/plain": [ "Numo::DFloat#shape=[3,3,3]\n", "[[[0.0617545, 0.373067, 0.794815], \n", " [0.201042, 0.116041, 0.344032], \n", " [0.539948, 0.737815, 0.165089]], \n", " [[0.0508827, 0.108065, 0.0687079], \n", " [0.904121, 0.478644, 0.342969], \n", " [0.164541, 0.74603, 0.138994]], \n", " [[0.411576, 0.292532, 0.869421], \n", " [0.0854984, 0.688965, 0.159977], \n", " [0.279215, 0.625155, 0.676329]]]" ] }, "execution_count": 13, "metadata": {}, "output_type": "execute_result" } ], "source": [ "z = Numo::DFloat.new(3,3,3).rand\n", "p z" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 13. Create a 10x10 array with random values and find the minimum and maximum values (★☆☆)\n", "\n", "\n", "Python:\n", "```python\n", "Z = np.random.random((10,10))\n", "Zmin, Zmax = Z.min(), Z.max()\n", "print(Zmin, Zmax)\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 14, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "0.0007664325967829586\n", "0.995590771731077\n" ] }, { "data": { "text/plain": [ "[0.0007664325967829586, 0.995590771731077]" ] }, "execution_count": 14, "metadata": {}, "output_type": "execute_result" } ], "source": [ "z = Numo::DFloat.new(10,10).rand\n", "zmin, zmax = z.minmax\n", "p zmin, zmax" ] }, { "cell_type": "markdown", "metadata": { "collapsed": true }, "source": [ "#### 14. Create a random vector of size 30 and find the mean value (★☆☆)\n", "\n", "\n", "Python:\n", "```python\n", "Z = np.random.random(30)\n", "m = Z.mean()\n", "print(m)\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 15, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "0.5609149765660713\n" ] }, { "data": { "text/plain": [ "0.5609149765660713" ] }, "execution_count": 15, "metadata": {}, "output_type": "execute_result" } ], "source": [ "z = Numo::DFloat.new(30).rand\n", "m = z.mean\n", "p m" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 15. Create a 2d array with 1 on the border and 0 inside (★☆☆)\n", "\n", "\n", "Python:\n", "```python\n", "Z = np.ones((10,10))\n", "Z[1:-1,1:-1] = 0\n", "print(Z)\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 16, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Numo::DFloat#shape=[10,10]\n", "[[1, 1, 1, 1, 1, 1, 1, 1, 1, 1], \n", " [1, 0, 0, 0, 0, 0, 0, 0, 0, 1], \n", " [1, 0, 0, 0, 0, 0, 0, 0, 0, 1], \n", " [1, 0, 0, 0, 0, 0, 0, 0, 0, 1], \n", " [1, 0, 0, 0, 0, 0, 0, 0, 0, 1], \n", " [1, 0, 0, 0, 0, 0, 0, 0, 0, 1], \n", " [1, 0, 0, 0, 0, 0, 0, 0, 0, 1], \n", " [1, 0, 0, 0, 0, 0, 0, 0, 0, 1], \n", " [1, 0, 0, 0, 0, 0, 0, 0, 0, 1], \n", " [1, 1, 1, 1, 1, 1, 1, 1, 1, 1]]\n" ] }, { "data": { "text/plain": [ "Numo::DFloat#shape=[10,10]\n", "[[1, 1, 1, 1, 1, 1, 1, 1, 1, 1], \n", " [1, 0, 0, 0, 0, 0, 0, 0, 0, 1], \n", " [1, 0, 0, 0, 0, 0, 0, 0, 0, 1], \n", " [1, 0, 0, 0, 0, 0, 0, 0, 0, 1], \n", " [1, 0, 0, 0, 0, 0, 0, 0, 0, 1], \n", " [1, 0, 0, 0, 0, 0, 0, 0, 0, 1], \n", " [1, 0, 0, 0, 0, 0, 0, 0, 0, 1], \n", " [1, 0, 0, 0, 0, 0, 0, 0, 0, 1], \n", " [1, 0, 0, 0, 0, 0, 0, 0, 0, 1], \n", " [1, 1, 1, 1, 1, 1, 1, 1, 1, 1]]" ] }, "execution_count": 16, "metadata": {}, "output_type": "execute_result" } ], "source": [ "z = Numo::DFloat.ones(10,10)\n", "z[1..-2,1..-2] = 0\n", "p z" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 16. How to add a border (filled with 0's) around an existing array? (★☆☆)\n", "\n", "\n", "Python:\n", "```python\n", "Z = np.ones((5,5))\n", "Z = np.pad(Z, pad_width=1, mode='constant', constant_values=0)\n", "print(Z)\n", "```\n", "\n", "Ruby:\n", "```ruby\n", "# todo: pad\n", "```" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 17. What is the result of the following expression? (★☆☆)\n", "\n", "\n", "Python:\n", "```python\n", "print(0 * np.nan)\n", "print(np.nan == np.nan)\n", "print(np.inf > np.nan)\n", "print(np.nan - np.nan)\n", "print(0.3 == 3 * 0.1)\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 17, "metadata": { "collapsed": false }, "outputs": [ { "data": { "text/plain": [ "false" ] }, "execution_count": 17, "metadata": {}, "output_type": "execute_result" } ], "source": [ "0 * Float::NAN\n", "Float::NAN == Float::NAN\n", "Float::INFINITY > Float::NAN\n", "Float::NAN - Float::NAN\n", "0.3 == 3 * 0.1" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 18. Create a 5x5 matrix with values 1,2,3,4 just below the diagonal (★☆☆)\n", "\n", "\n", "Python:\n", "```python\n", "Z = np.diag(1+np.arange(4),k=-1)\n", "print(Z)\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 18, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Numo::Int32#shape=[5,5]\n", "[[0, 0, 0, 0, 0], \n", " [1, 0, 0, 0, 0], \n", " [0, 2, 0, 0, 0], \n", " [0, 0, 3, 0, 0], \n", " [0, 0, 0, 4, 0]]\n" ] }, { "data": { "text/plain": [ "Numo::Int32#shape=[5,5]\n", "[[0, 0, 0, 0, 0], \n", " [1, 0, 0, 0, 0], \n", " [0, 2, 0, 0, 0], \n", " [0, 0, 3, 0, 0], \n", " [0, 0, 0, 4, 0]]" ] }, "execution_count": 18, "metadata": {}, "output_type": "execute_result" } ], "source": [ "z = Numo::Int32.zeros(5,5)\n", "z.diagonal(-1)[] = Numo::Int32[1..4]\n", "p z" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 19. Create a 8x8 matrix and fill it with a checkerboard pattern (★☆☆)\n", "\n", "\n", "Python:\n", "```python\n", "Z = np.zeros((8,8),dtype=int)\n", "Z[1::2,::2] = 1\n", "Z[::2,1::2] = 1\n", "print(Z)\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 19, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Numo::Int32#shape=[8,8]\n", "[[0, 1, 0, 1, 0, 1, 0, 1], \n", " [1, 0, 1, 0, 1, 0, 1, 0], \n", " [0, 1, 0, 1, 0, 1, 0, 1], \n", " [1, 0, 1, 0, 1, 0, 1, 0], \n", " [0, 1, 0, 1, 0, 1, 0, 1], \n", " [1, 0, 1, 0, 1, 0, 1, 0], \n", " [0, 1, 0, 1, 0, 1, 0, 1], \n", " [1, 0, 1, 0, 1, 0, 1, 0]]\n" ] }, { "data": { "text/plain": [ "Numo::Int32#shape=[8,8]\n", "[[0, 1, 0, 1, 0, 1, 0, 1], \n", " [1, 0, 1, 0, 1, 0, 1, 0], \n", " [0, 1, 0, 1, 0, 1, 0, 1], \n", " [1, 0, 1, 0, 1, 0, 1, 0], \n", " [0, 1, 0, 1, 0, 1, 0, 1], \n", " [1, 0, 1, 0, 1, 0, 1, 0], \n", " [0, 1, 0, 1, 0, 1, 0, 1], \n", " [1, 0, 1, 0, 1, 0, 1, 0]]" ] }, "execution_count": 19, "metadata": {}, "output_type": "execute_result" } ], "source": [ "# todo: rangewithstep\n", "x = Numo::Int32.new(1,8).seq\n", "y = Numo::Int32.new(8,1).seq\n", "z = (x+y)%2\n", "p z" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 20. Consider a (6,7,8) shape array, what is the index (x,y,z) of the 100th element?\n", "\n", "\n", "Python:\n", "```python\n", "print(np.unravel_index(100,(6,7,8)))\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 22, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "100\n" ] }, { "data": { "text/plain": [ "100" ] }, "execution_count": 22, "metadata": {}, "output_type": "execute_result" } ], "source": [ "# NArray allows unraveled index access\n", "z = Numo::Int32.new(6,7,8).seq\n", "p z[100]" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 21. Create a checkerboard 8x8 matrix using the tile function (★☆☆)\n", "\n", "\n", "Python:\n", "```python\n", "Z = np.tile( np.array([[0,1],[1,0]]), (4,4))\n", "print(Z)\n", "```\n", "\n", "Ruby:\n", "```ruby\n", "# todo: tile\n", "```" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 22. Normalize a 5x5 random matrix (★☆☆)\n", "\n", "\n", "Python:\n", "```python\n", "Z = np.random.random((5,5))\n", "Zmax, Zmin = Z.max(), Z.min()\n", "Z = (Z - Zmin)/(Zmax - Zmin)\n", "print(Z)\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 23, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Numo::DFloat#shape=[5,5]\n", "[[0.766088, 0.769435, 0.641326, 0.836021, 0.287046], \n", " [0.838608, 0.290923, 0.0930798, 0.235153, 0.57146], \n", " [0.167737, 0.548881, 1, 0.771149, 0.683695], \n", " [0.766882, 0.486607, 0.942667, 0, 0.45248], \n", " [0.801575, 0.23934, 0.267108, 0.536452, 0.382229]]\n" ] }, { "data": { "text/plain": [ "Numo::DFloat#shape=[5,5]\n", "[[0.766088, 0.769435, 0.641326, 0.836021, 0.287046], \n", " [0.838608, 0.290923, 0.0930798, 0.235153, 0.57146], \n", " [0.167737, 0.548881, 1, 0.771149, 0.683695], \n", " [0.766882, 0.486607, 0.942667, 0, 0.45248], \n", " [0.801575, 0.23934, 0.267108, 0.536452, 0.382229]]" ] }, "execution_count": 23, "metadata": {}, "output_type": "execute_result" } ], "source": [ "z = Numo::DFloat.new(5,5).rand\n", "zmin, zmax = z.minmax\n", "z = (z - zmin)/(zmax - zmin)\n", "p z" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 23. Create a custom dtype that describes a color as four unisgned bytes (RGBA) (★☆☆)\n", "\n", "\n", "Python:\n", "```python\n", "color = np.dtype([(\"r\", np.ubyte, 1),\n", " (\"g\", np.ubyte, 1),\n", " (\"b\", np.ubyte, 1),\n", " (\"a\", np.ubyte, 1)])\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 24, "metadata": { "collapsed": false }, "outputs": [ { "data": { "text/plain": [ "#" ] }, "execution_count": 24, "metadata": {}, "output_type": "execute_result" } ], "source": [ "# todo: record\n", "color = Numo::Struct.new do\n", " uint8 \"r\"\n", " uint8 \"g\"\n", " uint8 \"b\"\n", " uint8 \"a\"\n", "end" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 24. Multiply a 5x3 matrix by a 3x2 matrix (real matrix product) (★☆☆)\n", "\n", "\n", "Python:\n", "```python\n", "Z = np.dot(np.ones((5,3)), np.ones((3,2)))\n", "print(Z)\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 25, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Numo::DFloat#shape=[5,2]\n", "[[3, 3], \n", " [3, 3], \n", " [3, 3], \n", " [3, 3], \n", " [3, 3]]\n" ] }, { "data": { "text/plain": [ "Numo::DFloat#shape=[5,2]\n", "[[3, 3], \n", " [3, 3], \n", " [3, 3], \n", " [3, 3], \n", " [3, 3]]" ] }, "execution_count": 25, "metadata": {}, "output_type": "execute_result" } ], "source": [ "x = Numo::DFloat.ones(5,3)\n", "y = Numo::DFloat.ones(3,2)\n", "z = x.dot y\n", "p z" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 25. Given a 1D array, negate all elements which are between 3 and 8, in place. (★☆☆)\n", "\n", "\n", "Python:\n", "```python\n", "# Author: Evgeni Burovski\n", "\n", "Z = np.arange(11)\n", "Z[(3 < Z) & (Z <= 8)] *= -1\n", "print(Z)\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 27, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Numo::Int32#shape=[11]\n", "[0, 1, 2, 3, -4, -5, -6, -7, -8, 9, 10]\n" ] }, { "data": { "text/plain": [ "Numo::Int32#shape=[11]\n", "[0, 1, 2, 3, -4, -5, -6, -7, -8, 9, 10]" ] }, "execution_count": 27, "metadata": {}, "output_type": "execute_result" } ], "source": [ "z = Numo::Int32.new(11).seq\n", "z[(3 < z) & (z <= 8)] *= -1\n", "p z" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 26. What is the output of the following script? (★☆☆)\n", "\n", "\n", "Python:\n", "```python\n", "# Author: Jake VanderPlas\n", "\n", "print(sum(range(5),-1))\n", "from numpy import *\n", "print(sum(range(5),-1))\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 15, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "9\n", "10\n" ] }, { "data": { "text/plain": [ "10" ] }, "execution_count": 15, "metadata": {}, "output_type": "execute_result" } ], "source": [ "p [*0...5,-1].inject(:+)\n", "p Numo::Int32[0...5].sum(-1)" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 27. Consider an integer vector Z, which of these expressions are legal? (★☆☆)\n", "\n", "\n", "Python:\n", "```python\n", "Z = np.arange(5)\n", "Z**Z\n", "2 << Z >> 2\n", "Z <- Z\n", "1j*Z\n", "Z/1/1\n", "ZZ\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 16, "metadata": { "collapsed": false }, "outputs": [ { "ename": "TypeError", "evalue": "no implicit conversion of Numo::Int32 into Integer", "output_type": "error", "traceback": [ "\u001b[31mTypeError\u001b[0m: no implicit conversion of Numo::Int32 into Integer", "\u001b[37m(pry):51:in `<<'\u001b[0m", "\u001b[37m(pry):51:in `
'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/pry-0.10.4/lib/pry/pry_instance.rb:355:in `eval'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/pry-0.10.4/lib/pry/pry_instance.rb:355:in `evaluate_ruby'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/pry-0.10.4/lib/pry/pry_instance.rb:323:in `handle_line'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/pry-0.10.4/lib/pry/pry_instance.rb:243:in `block (2 levels) in eval'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/pry-0.10.4/lib/pry/pry_instance.rb:242:in `catch'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/pry-0.10.4/lib/pry/pry_instance.rb:242:in `block in eval'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/pry-0.10.4/lib/pry/pry_instance.rb:241:in `catch'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/pry-0.10.4/lib/pry/pry_instance.rb:241:in `eval'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/iruby-0.2.9/lib/iruby/backend.rb:65:in `eval'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/iruby-0.2.9/lib/iruby/backend.rb:12:in `eval'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/iruby-0.2.9/lib/iruby/kernel.rb:87:in `execute_request'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/iruby-0.2.9/lib/iruby/kernel.rb:47:in `dispatch'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/iruby-0.2.9/lib/iruby/kernel.rb:37:in `run'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/iruby-0.2.9/lib/iruby/command.rb:70:in `run_kernel'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/iruby-0.2.9/lib/iruby/command.rb:34:in `run'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/iruby-0.2.9/bin/iruby:5:in `'\u001b[0m", "\u001b[37m/usr/local/bin/iruby:22:in `load'\u001b[0m", "\u001b[37m/usr/local/bin/iruby:22:in `
'\u001b[0m" ] } ], "source": [ "z = Numo::Int32.new(5).seq\n", "z**z\n", "2 << z >> 2\n", "z <- z\n", "1i*z\n", "z/1/1\n", "zz" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 28. What are the result of the following expressions?\n", "\n", "\n", "Python:\n", "```python\n", "print(np.array(0) / np.array(0))\n", "print(np.array(0) // np.array(0))\n", "print(np.array([np.nan]).astype(int).astype(float))\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 17, "metadata": { "collapsed": false }, "outputs": [ { "ename": "ZeroDivisionError", "evalue": "error in NArray operation", "output_type": "error", "traceback": [ "\u001b[31mZeroDivisionError\u001b[0m: error in NArray operation", "\u001b[37m(pry):56:in `/'\u001b[0m", "\u001b[37m(pry):56:in `
'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/pry-0.10.4/lib/pry/pry_instance.rb:355:in `eval'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/pry-0.10.4/lib/pry/pry_instance.rb:355:in `evaluate_ruby'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/pry-0.10.4/lib/pry/pry_instance.rb:323:in `handle_line'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/pry-0.10.4/lib/pry/pry_instance.rb:243:in `block (2 levels) in eval'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/pry-0.10.4/lib/pry/pry_instance.rb:242:in `catch'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/pry-0.10.4/lib/pry/pry_instance.rb:242:in `block in eval'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/pry-0.10.4/lib/pry/pry_instance.rb:241:in `catch'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/pry-0.10.4/lib/pry/pry_instance.rb:241:in `eval'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/iruby-0.2.9/lib/iruby/backend.rb:65:in `eval'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/iruby-0.2.9/lib/iruby/backend.rb:12:in `eval'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/iruby-0.2.9/lib/iruby/kernel.rb:87:in `execute_request'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/iruby-0.2.9/lib/iruby/kernel.rb:47:in `dispatch'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/iruby-0.2.9/lib/iruby/kernel.rb:37:in `run'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/iruby-0.2.9/lib/iruby/command.rb:70:in `run_kernel'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/iruby-0.2.9/lib/iruby/command.rb:34:in `run'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/iruby-0.2.9/bin/iruby:5:in `'\u001b[0m", "\u001b[37m/usr/local/bin/iruby:22:in `load'\u001b[0m", "\u001b[37m/usr/local/bin/iruby:22:in `
'\u001b[0m" ] } ], "source": [ "p Numo::Int32[0] / Numo::Int32[0]\n", "p Numo::DFloat[Float::NAN].cast_to(Numo::Int32).cast_to(Numo::DFloat)" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 29. How to round away from zero a float array ? (★☆☆)\n", "\n", "\n", "Python:\n", "```python\n", "# Author: Charles R Harris\n", "\n", "Z = np.random.uniform(-10,+10,10)\n", "print (np.trunc(Z + np.copysign(0.5, Z)))\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 18, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Numo::DFloat#shape=[10]\n", "[-7, -0, 10, -5, -7, -1, -10, -3, -1, -5]\n" ] }, { "data": { "text/plain": [ "Numo::DFloat#shape=[10]\n", "[-7, -0, 10, -5, -7, -1, -10, -3, -1, -5]" ] }, "execution_count": 18, "metadata": {}, "output_type": "execute_result" } ], "source": [ "z = Numo::DFloat.new(10).rand(-10,+10)\n", "p (z + (0.5*z.sign)).trunc\n", "# todo: copysign" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 30. How to find common values between two arrays? (★☆☆)\n", "\n", "\n", "Python:\n", "```python\n", "Z1 = np.random.randint(0,10,10)\n", "Z2 = np.random.randint(0,10,10)\n", "print(np.intersect1d(Z1,Z2))\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 19, "metadata": { "collapsed": true }, "outputs": [], "source": [ "# todo: intersect1d" ] }, { "cell_type": "markdown", "metadata": { "collapsed": true }, "source": [ "#### 35. How to compute ((A+B)\\*(-A/2)) in place (without copy)? (★★☆)\n", "\n", "\n", "Python:\n", "```python\n", "A = np.ones(3)*1\n", "B = np.ones(3)*2\n", "np.add(A,B,out=B)\n", "np.divide(A,2,out=A)\n", "np.negative(A,out=A)\n", "np.multiply(A,B,out=A)\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 3, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Numo::DFloat#shape=[3]\n", "[-1.5, -1.5, -1.5]\n", "Numo::DFloat#shape=[3]\n", "[-1.5, -1.5, -1.5]\n" ] }, { "data": { "text/plain": [ "Numo::DFloat#shape=[3]\n", "[-1.5, -1.5, -1.5]" ] }, "execution_count": 3, "metadata": {}, "output_type": "execute_result" } ], "source": [ "a = Numo::DFloat.new(3).fill(1)\n", "b = Numo::DFloat.new(3).fill(2)\n", "p (a+b)*(-a/2)\n", "(a+b.inplace)*(-a.inplace/2)\n", "p b" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 36. Extract the integer part of a random array using 5 different methods (★★☆)\n", "\n", "\n", "Python:\n", "```python\n", "Z = np.random.uniform(0,10,10)\n", "\n", "print (Z - Z%1)\n", "print (np.floor(Z))\n", "print (np.ceil(Z)-1)\n", "print (Z.astype(int))\n", "print (np.trunc(Z))\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 4, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Numo::DFloat#shape=[10]\n", "[0, 3, 7, 2, 1, 3, 5, 7, 1, 0]\n", "Numo::DFloat#shape=[10]\n", "[0, 3, 7, 2, 1, 3, 5, 7, 1, 0]\n", "Numo::DFloat#shape=[10]\n", "[0, 3, 7, 2, 1, 3, 5, 7, 1, 0]\n", "Numo::Int32#shape=[10]\n", "[0, 3, 7, 2, 1, 3, 5, 7, 1, 0]\n", "Numo::DFloat#shape=[10]\n", "[0, 3, 7, 2, 1, 3, 5, 7, 1, 0]\n" ] }, { "data": { "text/plain": [ "Numo::DFloat#shape=[10]\n", "[0, 3, 7, 2, 1, 3, 5, 7, 1, 0]" ] }, "execution_count": 4, "metadata": {}, "output_type": "execute_result" } ], "source": [ "z = Numo::DFloat.new(10).rand(10)\n", "\n", "p z - z%1\n", "p z.floor\n", "p z.ceil - 1\n", "p z.cast_to(Numo::Int32)\n", "p z.trunc" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 37. Create a 5x5 matrix with row values ranging from 0 to 4 (★★☆)\n", "\n", "\n", "Python:\n", "```python\n", "Z = np.zeros((5,5))\n", "Z += np.arange(5)\n", "print(Z)\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 5, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Numo::DFloat#shape=[5,5]\n", "[[0, 1, 2, 3, 4], \n", " [0, 1, 2, 3, 4], \n", " [0, 1, 2, 3, 4], \n", " [0, 1, 2, 3, 4], \n", " [0, 1, 2, 3, 4]]\n" ] }, { "data": { "text/plain": [ "Numo::DFloat#shape=[5,5]\n", "[[0, 1, 2, 3, 4], \n", " [0, 1, 2, 3, 4], \n", " [0, 1, 2, 3, 4], \n", " [0, 1, 2, 3, 4], \n", " [0, 1, 2, 3, 4]]" ] }, "execution_count": 5, "metadata": {}, "output_type": "execute_result" } ], "source": [ "z = Numo::DFloat.zeros(5,5)\n", "z += Numo::Int32.new(5).seq\n", "p z" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 39. Create a vector of size 10 with values ranging from 0 to 1, both excluded (★★☆)\n", "\n", "\n", "Python:\n", "```python\n", "Z = np.linspace(0,1,12,endpoint=True)[1:-1]\n", "print(Z)\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 6, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Numo::DFloat(view)#shape=[10]\n", "[0.0909091, 0.181818, 0.272727, 0.363636, 0.454545, 0.545455, 0.636364, ...]\n" ] }, { "data": { "text/plain": [ "Numo::DFloat(view)#shape=[10]\n", "[0.0909091, 0.181818, 0.272727, 0.363636, 0.454545, 0.545455, 0.636364, ...]" ] }, "execution_count": 6, "metadata": {}, "output_type": "execute_result" } ], "source": [ "z = Numo::DFloat.linspace(0,1,12)[1..-2]\n", "p z" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 40. Create a random vector of size 10 and sort it (★★☆)\n", "\n", "\n", "Python:\n", "```python\n", "Z = np.random.random(10)\n", "Z.sort()\n", "print(Z)\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 7, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Numo::DFloat#shape=[10]\n", "[0.0687079, 0.108065, 0.138994, 0.164541, 0.292532, 0.342969, 0.411576, ...]\n" ] }, { "data": { "text/plain": [ "Numo::DFloat#shape=[10]\n", "[0.0687079, 0.108065, 0.138994, 0.164541, 0.292532, 0.342969, 0.411576, ...]" ] }, "execution_count": 7, "metadata": {}, "output_type": "execute_result" } ], "source": [ "z = Numo::DFloat.new(10).rand\n", "z = z.sort\n", "p z" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 43. Make an array immutable (read-only) (★★☆)\n", "\n", "\n", "Python:\n", "```python\n", "Z = np.zeros(10)\n", "Z.flags.writeable = False\n", "Z[0] = 1\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 3, "metadata": { "collapsed": false }, "outputs": [ { "ename": "RuntimeError", "evalue": "cannot write to frozen NArray.", "output_type": "error", "traceback": [ "\u001b[31mRuntimeError\u001b[0m: cannot write to frozen NArray.", "\u001b[37m(pry):7:in `[]='\u001b[0m", "\u001b[37m(pry):7:in `
'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/pry-0.10.4/lib/pry/pry_instance.rb:355:in `eval'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/pry-0.10.4/lib/pry/pry_instance.rb:355:in `evaluate_ruby'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/pry-0.10.4/lib/pry/pry_instance.rb:323:in `handle_line'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/pry-0.10.4/lib/pry/pry_instance.rb:243:in `block (2 levels) in eval'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/pry-0.10.4/lib/pry/pry_instance.rb:242:in `catch'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/pry-0.10.4/lib/pry/pry_instance.rb:242:in `block in eval'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/pry-0.10.4/lib/pry/pry_instance.rb:241:in `catch'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/pry-0.10.4/lib/pry/pry_instance.rb:241:in `eval'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/iruby-0.2.9/lib/iruby/backend.rb:65:in `eval'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/iruby-0.2.9/lib/iruby/backend.rb:12:in `eval'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/iruby-0.2.9/lib/iruby/kernel.rb:87:in `execute_request'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/iruby-0.2.9/lib/iruby/kernel.rb:47:in `dispatch'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/iruby-0.2.9/lib/iruby/kernel.rb:37:in `run'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/iruby-0.2.9/lib/iruby/command.rb:70:in `run_kernel'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/iruby-0.2.9/lib/iruby/command.rb:34:in `run'\u001b[0m", "\u001b[37m/var/lib/gems/2.1.0/gems/iruby-0.2.9/bin/iruby:5:in `'\u001b[0m", "\u001b[37m/usr/local/bin/iruby:22:in `load'\u001b[0m", "\u001b[37m/usr/local/bin/iruby:22:in `
'\u001b[0m" ] } ], "source": [ "z = Numo::DFloat.zeros(10)\n", "z.freeze\n", "z[0] = 1" ] }, { "cell_type": "markdown", "metadata": { "collapsed": true }, "source": [ "#### 44. Consider a random 10x2 matrix representing cartesian coordinates, convert them to polar coordinates (★★☆)\n", "\n", "\n", "Python:\n", "```python\n", "Z = np.random.random((10,2))\n", "X,Y = Z[:,0], Z[:,1]\n", "R = np.sqrt(X**2+Y**2)\n", "T = np.arctan2(Y,X)\n", "print(R)\n", "print(T)\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 4, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Numo::DFloat#shape=[10]\n", "[0.378143, 0.819847, 0.363075, 0.914284, 0.172752, 0.128057, 1.023, ...]\n", "Numo::DFloat#shape=[10]\n", "[1.40675, 0.247746, 1.24548, 0.939032, 0.298976, 0.566331, 0.486892, ...]\n" ] }, { "data": { "text/plain": [ "Numo::DFloat#shape=[10]\n", "[1.40675, 0.247746, 1.24548, 0.939032, 0.298976, 0.566331, 0.486892, ...]" ] }, "execution_count": 4, "metadata": {}, "output_type": "execute_result" } ], "source": [ "z = Numo::DFloat.new(10,2).rand\n", "x,y = z[true,0], z[true,1]\n", "r = Numo::NMath.sqrt(x**2+y**2)\n", "t = Numo::NMath.atan2(y,x)\n", "p r\n", "p t" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 45. Create random vector of size 10 and replace the maximum value by 0 (★★☆)\n", "\n", "\n", "Python:\n", "```python\n", "Z = np.random.random(10)\n", "Z[Z.argmax()] = 0\n", "print(Z)\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 5, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Numo::DFloat#shape=[10]\n", "[0, 0.0854984, 0.688965, 0.159977, 0.279215, 0.625155, 0.676329, ...]\n" ] }, { "data": { "text/plain": [ "Numo::DFloat#shape=[10]\n", "[0, 0.0854984, 0.688965, 0.159977, 0.279215, 0.625155, 0.676329, ...]" ] }, "execution_count": 5, "metadata": {}, "output_type": "execute_result" } ], "source": [ "z = Numo::DFloat.new(10).rand\n", "z[z.max_index] = 0\n", "p z" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 50. How to find the closest value (to a given scalar) in an array? (★★☆)\n", "\n", "\n", "Python:\n", "```python\n", "Z = np.arange(100)\n", "v = np.random.uniform(0,100)\n", "index = (np.abs(Z-v)).argmin()\n", "print(Z[index])\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 3, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "86\n" ] }, { "data": { "text/plain": [ "86" ] }, "execution_count": 3, "metadata": {}, "output_type": "execute_result" } ], "source": [ "z = Numo::Int32.new(100).seq\n", "v = rand*100\n", "index = (z-v).abs.min_index\n", "p z[index]" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 54. How to read the following file? (★★☆)\n", "\n", "\n", "Python:\n", "```python\n", "from io import StringIO\n", "\n", "# Fake file\n", "s = StringIO(\"\"\"1, 2, 3, 4, 5\\n\n", " 6, , , 7, 8\\n\n", " , , 9,10,11\\n\"\"\")\n", "Z = np.genfromtxt(s, delimiter=\",\", dtype=np.int)\n", "print(Z)\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 4, "metadata": { "collapsed": false }, "outputs": [ { "data": { "text/plain": [ "Numo::DFloat#shape=[3,5]\n", "[[1, 2, 3, 4, 5], \n", " [6, nan, nan, 7, 8], \n", " [nan, nan, 9, 10, 11]]" ] }, "execution_count": 4, "metadata": {}, "output_type": "execute_result" } ], "source": [ "require \"stringio\"\n", "s = StringIO.new(\"1, 2, 3, 4, 5\n", " 6, , , 7, 8\n", " , , 9,10,11\")\n", "z = Numo::NArray[*s.readlines.map{|l| l.split(\",\").map{|x| x.strip.empty? ? Float::NAN : x.to_f}}]" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 55. What is the equivalent of enumerate for numpy arrays? (★★☆)\n", "\n", "\n", "Python:\n", "```python\n", "Z = np.arange(9).reshape(3,3)\n", "for index, value in np.ndenumerate(Z):\n", " print(index, value)\n", "for index in np.ndindex(Z.shape):\n", " print(index, Z[index])\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 5, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "[[0, 0], 0]\n", "[[0, 1], 1]\n", "[[0, 2], 2]\n", "[[1, 0], 3]\n", "[[1, 1], 4]\n", "[[1, 2], 5]\n", "[[2, 0], 6]\n", "[[2, 1], 7]\n", "[[2, 2], 8]\n" ] }, { "data": { "text/plain": [ "Numo::Int32#shape=[3,3]\n", "[[0, 1, 2], \n", " [3, 4, 5], \n", " [6, 7, 8]]" ] }, "execution_count": 5, "metadata": {}, "output_type": "execute_result" } ], "source": [ "z = Numo::Int32.new(3,3).seq\n", "z.each_with_index{|x,*i| p [i,x]}" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 56. Generate a generic 2D Gaussian-like array (★★☆)\n", "\n", "\n", "Python:\n", "```python\n", "X, Y = np.meshgrid(np.linspace(-1,1,10), np.linspace(-1,1,10))\n", "D = np.sqrt(X*X+Y*Y)\n", "sigma, mu = 1.0, 0.0\n", "G = np.exp(-( (D-mu)**2 / ( 2.0 * sigma**2 ) ) )\n", "print(G)\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 6, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Numo::DFloat#shape=[10,10]\n", "[[0.367879, 0.448221, 0.519795, 0.573753, 0.602798, 0.602798, 0.573753, ...], \n", " [0.448221, 0.546108, 0.633313, 0.699056, 0.734444, 0.734444, 0.699056, ...], \n", " [0.519795, 0.633313, 0.734444, 0.810684, 0.851723, 0.851723, 0.810684, ...], \n", " [0.573753, 0.699056, 0.810684, 0.894839, 0.940138, 0.940138, 0.894839, ...], \n", " [0.602798, 0.734444, 0.851723, 0.940138, 0.98773, 0.98773, 0.940138, ...], \n", " [0.602798, 0.734444, 0.851723, 0.940138, 0.98773, 0.98773, 0.940138, ...], \n", " [0.573753, 0.699056, 0.810684, 0.894839, 0.940138, 0.940138, 0.894839, ...], \n", " [0.519795, 0.633313, 0.734444, 0.810684, 0.851723, 0.851723, 0.810684, ...], \n", " [0.448221, 0.546108, 0.633313, 0.699056, 0.734444, 0.734444, 0.699056, ...], \n", " [0.367879, 0.448221, 0.519795, 0.573753, 0.602798, 0.602798, 0.573753, ...]]\n" ] }, { "data": { "text/plain": [ "Numo::DFloat#shape=[10,10]\n", "[[0.367879, 0.448221, 0.519795, 0.573753, 0.602798, 0.602798, 0.573753, ...], \n", " [0.448221, 0.546108, 0.633313, 0.699056, 0.734444, 0.734444, 0.699056, ...], \n", " [0.519795, 0.633313, 0.734444, 0.810684, 0.851723, 0.851723, 0.810684, ...], \n", " [0.573753, 0.699056, 0.810684, 0.894839, 0.940138, 0.940138, 0.894839, ...], \n", " [0.602798, 0.734444, 0.851723, 0.940138, 0.98773, 0.98773, 0.940138, ...], \n", " [0.602798, 0.734444, 0.851723, 0.940138, 0.98773, 0.98773, 0.940138, ...], \n", " [0.573753, 0.699056, 0.810684, 0.894839, 0.940138, 0.940138, 0.894839, ...], \n", " [0.519795, 0.633313, 0.734444, 0.810684, 0.851723, 0.851723, 0.810684, ...], \n", " [0.448221, 0.546108, 0.633313, 0.699056, 0.734444, 0.734444, 0.699056, ...], \n", " [0.367879, 0.448221, 0.519795, 0.573753, 0.602798, 0.602798, 0.573753, ...]]" ] }, "execution_count": 6, "metadata": {}, "output_type": "execute_result" } ], "source": [ "x = Numo::DFloat.linspace(-1,1,10)\n", "y = Numo::DFloat.linspace(-1,1,10).expand_dims(1)\n", "d = Numo::NMath.sqrt(x*x+y*y)\n", "sigma, mu = 1.0, 0.0\n", "g = Numo::NMath.exp(-( (d-mu)**2 / ( 2.0 * sigma**2 ) ) )\n", "p g" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 58. Subtract the mean of each row of a matrix (★★☆)\n", "\n", "\n", "Python:\n", "```python\n", "# Author: Warren Weckesser\n", "\n", "X = np.random.rand(5, 10)\n", "\n", "# Recent versions of numpy\n", "Y = X - X.mean(axis=1, keepdims=True)\n", "\n", "# Older versions of numpy\n", "Y = X - X.mean(axis=1).reshape(-1, 1)\n", "\n", "print(Y)\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 7, "metadata": { "collapsed": false }, "outputs": [ { "data": { "text/plain": [ "Numo::DFloat#shape=[5,10]\n", "[[-0.276694, 0.0346182, 0.456367, -0.137406, -0.222408, 0.00558358, ...], \n", " [-0.257554, -0.29691, 0.538503, 0.113026, -0.0226486, -0.201077, ...], \n", " [0.49633, -0.287593, 0.315874, -0.213114, -0.0938766, 0.252064, 0.303238, ...], \n", " [-0.0359266, -0.116908, 0.508809, 0.423319, -0.309937, -0.404925, ...], \n", " [-0.168796, -0.165424, 0.464085, -0.0274508, 0.0251129, 0.0339921, ...]]" ] }, "execution_count": 7, "metadata": {}, "output_type": "execute_result" } ], "source": [ "x = Numo::DFloat.new(5, 10).rand\n", "y = x - x.mean(1).expand_dims(1)" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 59. How to I sort an array by the nth column? (★★☆)\n", "\n", "\n", "Python:\n", "```python\n", "# Author: Steve Tjoa\n", "\n", "Z = np.random.randint(0,10,(3,3))\n", "print(Z)\n", "print(Z[Z[:,1].argsort()])\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 8, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Numo::Int32#shape=[3,3]\n", "[[7, 3, 0], \n", " [2, 5, 0], \n", " [2, 7, 7]]\n", "Numo::Int32(view)#shape=[3,3]\n", "[[7, 3, 0], \n", " [2, 5, 0], \n", " [2, 7, 7]]\n" ] }, { "data": { "text/plain": [ "Numo::Int32(view)#shape=[3,3]\n", "[[7, 3, 0], \n", " [2, 5, 0], \n", " [2, 7, 7]]" ] }, "execution_count": 8, "metadata": {}, "output_type": "execute_result" } ], "source": [ "z = Numo::Int32.new(3,3).rand(10)\n", "p z\n", "p z[z[true,1].sort_index,true]" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 60. How to tell if a given 2D array has null columns? (★★☆)\n", "\n", "\n", "Python:\n", "```python\n", "# Author: Warren Weckesser\n", "\n", "Z = np.random.randint(0,3,(3,10))\n", "print((~Z.any(axis=0)).any())\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 2, "metadata": { "collapsed": false }, "outputs": [ { "data": { "text/plain": [ "true" ] }, "execution_count": 2, "metadata": {}, "output_type": "execute_result" } ], "source": [ "z = Numo::Int32.new(3,10).rand(3)\n", "(~z.ne(0).any?(0)).any?" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 61. Find the nearest value from a given value in an array (★★☆)\n", "\n", "\n", "Python:\n", "```python\n", "Z = np.random.uniform(0,1,10)\n", "z = 0.5\n", "m = Z.flat[np.abs(Z - z).argmin()]\n", "print(m)\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 3, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "0.4115760120668863\n" ] }, { "data": { "text/plain": [ "0.4115760120668863" ] }, "execution_count": 3, "metadata": {}, "output_type": "execute_result" } ], "source": [ "z = Numo::DFloat.new(10).rand\n", "x = 0.5\n", "m = z[(z - x).abs.min_index]\n", "p m" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 66. Considering a (w,h,3) image of (dtype=ubyte), compute the number of unique colors (★★★)\n", "\n", "\n", "Python:\n", "```python\n", "# Author: Nadav Horesh\n", "\n", "w,h = 16,16\n", "I = np.random.randint(0,2,(h,w,3)).astype(np.ubyte)\n", "F = I[...,0]*256*256 + I[...,1]*256 +I[...,2]\n", "n = len(np.unique(F))\n", "print(np.unique(I))\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 4, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "[0, 1, 256, 257, 65536, 65793, 65792, 65537]\n" ] }, { "data": { "text/plain": [ "[0, 1, 256, 257, 65536, 65793, 65792, 65537]" ] }, "execution_count": 4, "metadata": {}, "output_type": "execute_result" } ], "source": [ "# todo: unique\n", "w,h = 16,16\n", "i = Numo::UInt32.new(h,w,3).rand(2)\n", "f = i[false,0]*256*256 + i[false,1]*256 +i[false,2]\n", "p f.flatten.sort.to_a.uniq" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 67. Considering a four dimensions array, how to get sum over the last two axis at once? (★★★)\n", "\n", "\n", "Python:\n", "```python\n", "A = np.random.randint(0,10,(3,4,3,4))\n", "sum = A.reshape(A.shape[:-2] + (-1,)).sum(axis=-1)\n", "print(sum)\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 5, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Numo::Int32#shape=[3,4]\n", "[[80, 51, 48, 48], \n", " [51, 58, 56, 43], \n", " [59, 61, 73, 54]]\n" ] }, { "data": { "text/plain": [ "Numo::Int32#shape=[3,4]\n", "[[80, 51, 48, 48], \n", " [51, 58, 56, 43], \n", " [59, 61, 73, 54]]" ] }, "execution_count": 5, "metadata": {}, "output_type": "execute_result" } ], "source": [ "a = Numo::Int32.new(3,4,3,4).rand(10)\n", "sum = a.sum(-2,-1)\n", "p sum" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 68. Considering a one-dimensional vector D, how to compute means of subsets of D using a vector S of same size describing subset indices? (★★★)\n", "\n", "\n", "Python:\n", "```python\n", "# Author: Jaime Fernández del Río\n", "\n", "D = np.random.uniform(0,1,100)\n", "S = np.random.randint(0,10,100)\n", "D_sums = np.bincount(S, weights=D)\n", "D_counts = np.bincount(S)\n", "D_means = D_sums / D_counts\n", "print(D_means)\n", "```\n", "\n", "Ruby:\n", "```ruby\n", "# todo: bincount\n", "```" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 69. How to get the diagonal of a dot product? (★★★)\n", "\n", "\n", "Python:\n", "```python\n", "# Author: Mathieu Blondel\n", "\n", "A = np.random.uniform(0,1,(5,5))\n", "B = np.random.uniform(0,1,(5,5))\n", "\n", "# Slow version\n", "np.diag(np.dot(A, B))\n", "\n", "# Fast version\n", "np.sum(A * B.T, axis=1)\n", "\n", "# Faster version\n", "np.einsum(\"ij,ji->i\", A, B)\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 6, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Numo::DFloat#shape=[3]\n", "[15, 54, 111]\n" ] }, { "data": { "text/plain": [ "Numo::DFloat#shape=[3]\n", "[15, 54, 111]" ] }, "execution_count": 6, "metadata": {}, "output_type": "execute_result" } ], "source": [ "a = Numo::DFloat.new(3,3).seq\n", "b = Numo::DFloat.new(3,3).seq\n", "p a.mulsum(b.transpose,1)\n", "# speed?" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 70. Consider the vector \\[1, 2, 3, 4, 5\\], how to build a new vector with 3 consecutive zeros interleaved between each value? (★★★)\n", "\n", "\n", "Python:\n", "```python\n", "# Author: Warren Weckesser\n", "\n", "Z = np.array([1,2,3,4,5])\n", "nz = 3\n", "Z0 = np.zeros(len(Z) + (len(Z)-1)*(nz))\n", "Z0[::nz+1] = Z\n", "print(Z0)\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 7, "metadata": { "collapsed": false }, "outputs": [ { "data": { "text/plain": [ "Numo::Int32#shape=[17]\n", "[0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]" ] }, "execution_count": 7, "metadata": {}, "output_type": "execute_result" } ], "source": [ "z = Numo::NArray[1,2,3,4,5]\n", "nz = 3\n", "z0 = Numo::Int32.zeros(z.size + (z.size-1)*(nz))\n", "# todo: rangewithstep\n", "# z0[(0..-1).step(nz+1)] = z\n", "# p z0" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 71. Consider an array of dimension (5,5,3), how to mulitply it by an array with dimensions (5,5)? (★★★)\n", "\n", "\n", "Python:\n", "```python\n", "A = np.ones((5,5,3))\n", "B = 2*np.ones((5,5))\n", "print(A * B[:,:,None])\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 8, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Numo::Int32#shape=[5,5,3]\n", "[[[2, 2, 2], \n", " [2, 2, 2], \n", " [2, 2, 2], \n", " [2, 2, 2], \n", " [2, 2, 2]], \n", " [[2, 2, 2], \n", " [2, 2, 2], \n", " [2, 2, 2], \n", " [2, 2, 2], \n", " [2, 2, 2]], \n", " [[2, 2, 2], \n", " [2, 2, 2], \n", " [2, 2, 2], \n", " [2, 2, 2], \n", " [2, 2, 2]], \n", " [[2, 2, 2], \n", " [2, 2, 2], \n", " [2, 2, 2], \n", " [2, 2, 2], \n", " [2, 2, 2]], \n", " ...\n" ] }, { "data": { "text/plain": [ "Numo::Int32#shape=[5,5,3]\n", "[[[2, 2, 2], \n", " [2, 2, 2], \n", " [2, 2, 2], \n", " [2, 2, 2], \n", " [2, 2, 2]], \n", " [[2, 2, 2], \n", " [2, 2, 2], \n", " [2, 2, 2], \n", " [2, 2, 2], \n", " [2, 2, 2]], \n", " [[2, 2, 2], \n", " [2, 2, 2], \n", " [2, 2, 2], \n", " [2, 2, 2], \n", " [2, 2, 2]], \n", " [[2, 2, 2], \n", " [2, 2, 2], \n", " [2, 2, 2], \n", " [2, 2, 2], \n", " [2, 2, 2]], \n", " ..." ] }, "execution_count": 8, "metadata": {}, "output_type": "execute_result" } ], "source": [ "a = Numo::Int32.ones(5,5,3)\n", "b = Numo::Int32.new(5,5).fill(2)\n", "p a * b[:*,:*,:-]" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 72. How to swap two rows of an array? (★★★)\n", "\n", "\n", "Python:\n", "```python\n", "# Author: Eelco Hoogendoorn\n", "\n", "A = np.arange(25).reshape(5,5)\n", "A[[0,1]] = A[[1,0]]\n", "print(A)\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 9, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Numo::Int32#shape=[5,5]\n", "[[5, 6, 7, 8, 9], \n", " [0, 1, 2, 3, 4], \n", " [10, 11, 12, 13, 14], \n", " [15, 16, 17, 18, 19], \n", " [20, 21, 22, 23, 24]]\n" ] }, { "data": { "text/plain": [ "Numo::Int32#shape=[5,5]\n", "[[5, 6, 7, 8, 9], \n", " [0, 1, 2, 3, 4], \n", " [10, 11, 12, 13, 14], \n", " [15, 16, 17, 18, 19], \n", " [20, 21, 22, 23, 24]]" ] }, "execution_count": 9, "metadata": {}, "output_type": "execute_result" } ], "source": [ "a = Numo::Int32.new(5,5).seq\n", "a[[0,1],true] = a[[1,0],true].copy\n", "p a\n", "# todo: identity check between read/write array" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 73. Consider a set of 10 triplets describing 10 triangles (with shared vertices), find the set of unique line segments composing all the triangles (★★★)\n", "\n", "\n", "Python:\n", "```python\n", "# Author: Nicolas P. Rougier\n", "\n", "faces = np.random.randint(0,100,(10,3))\n", "F = np.roll(faces.repeat(2,axis=1),-1,axis=1)\n", "F = F.reshape(len(F)*3,2)\n", "F = np.sort(F,axis=1)\n", "G = F.view( dtype=[('p0',F.dtype),('p1',F.dtype)] )\n", "G = np.unique(G)\n", "print(G)\n", "```\n", "\n", "Ruby:\n", "```ruby\n", "# todo: roll\n", "```" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 74. Given an array C that is a bincount, how to produce an array A such that np.bincount(A) == C? (★★★)\n", "\n", "\n", "Python:\n", "```python\n", "# Author: Jaime Fernández del Río\n", "\n", "C = np.bincount([1,1,2,3,4,4,6])\n", "A = np.repeat(np.arange(len(C)), C)\n", "print(A)\n", "```\n", "\n", "Ruby:\n", "```ruby\n", "# todo: bincount, repeat\n", "```" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 75. How to compute averages using a sliding window over an array? (★★★)\n", "\n", "\n", "Python:\n", "```python\n", "# Author: Jaime Fernández del Río\n", "\n", "def moving_average(a, n=3) :\n", " ret = np.cumsum(a, dtype=float)\n", " ret[n:] = ret[n:] - ret[:-n]\n", " return ret[n - 1:] / n\n", "Z = np.arange(20)\n", "print(moving_average(Z, n=3))\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 2, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Numo::DFloat#shape=[18]\n", "[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18]\n" ] }, { "data": { "text/plain": [ "Numo::DFloat#shape=[18]\n", "[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18]" ] }, "execution_count": 2, "metadata": {}, "output_type": "execute_result" } ], "source": [ "def moving_average(a, n=3)\n", " ret = a.cumsum\n", " ret[n..-1] = ret[n..-1] - ret[0..-n-1]\n", " ret[n-1..-1] / n\n", "end\n", "z = Numo::DFloat.new(20).seq\n", "p moving_average(z, 3)" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 76. Consider a one-dimensional array Z, build a two-dimensional array whose first row is (Z\\[0\\],Z\\[1\\],Z\\[2\\]) and each subsequent row is shifted by 1 (last row should be (Z\\[-3\\],Z\\[-2\\],Z\\[-1\\]) (★★★)\n", "\n", "\n", "Python:\n", "```python\n", "# Author: Joe Kington / Erik Rigtorp\n", "from numpy.lib import stride_tricks\n", "\n", "def rolling(a, window):\n", " shape = (a.size - window + 1, window)\n", " strides = (a.itemsize, a.itemsize)\n", " return stride_tricks.as_strided(a, shape=shape, strides=strides)\n", "Z = rolling(np.arange(10), 3)\n", "print(Z)\n", "```\n", "\n", "Ruby:\n", "```ruby\n", "# no module: stride_tricks\n", "```" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 77. How to negate a boolean, or to change the sign of a float inplace? (★★★)\n", "\n", "\n", "Python:\n", "```python\n", "# Author: Nathaniel J. Smith\n", "\n", "Z = np.random.randint(0,2,100)\n", "np.logical_not(Z, out=Z)\n", "\n", "Z = np.random.uniform(-1.0,1.0,100)\n", "np.negative(Z, out=Z)\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 3, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Numo::Int32#shape=[100]\n", "[0, 0, 0, 1, 0, 0, 1, 1, 1, 0, 0, 0, 0, 1, 0, 1, 1, 1, 0, 1, 1, 0, 1, 1, 0, ...]\n", "Numo::Int32#shape=[100]\n", "[1, 1, 1, 0, 1, 1, 0, 0, 0, 1, 1, 1, 1, 0, 1, 0, 0, 0, 1, 0, 0, 1, 0, 0, 1, ...]\n", "Numo::DFloat#shape=[100]\n", "[-0.00105903, 0.140595, -0.671175, 0.782669, 0.406039, -0.54762, 0.86998, ...]\n", "Numo::DFloat#shape=[100]\n", "[0.00105903, -0.140595, 0.671175, -0.782669, -0.406039, 0.54762, -0.86998, ...]\n" ] }, { "data": { "text/plain": [ "Numo::DFloat#shape=[100]\n", "[0.00105903, -0.140595, 0.671175, -0.782669, -0.406039, 0.54762, -0.86998, ...]" ] }, "execution_count": 3, "metadata": {}, "output_type": "execute_result" } ], "source": [ "# todo: logical_not\n", "z = Numo::Int32.new(100).rand(2)\n", "p z\n", "z.inplace ^ 1\n", "p z\n", "\n", "z = Numo::DFloat.new(100).rand(-1,1)\n", "p z\n", "-z.inplace\n", "p z" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 78. Consider 2 sets of points P0,P1 describing lines (2d) and a point p, how to compute distance from p to each line i (P0\\[i\\],P1\\[i\\])? (★★★)\n", "\n", "\n", "Python:\n", "```python\n", "def distance(P0, P1, p):\n", " T = P1 - P0\n", " L = (T**2).sum(axis=1)\n", " U = -((P0[:,0]-p[...,0])*T[:,0] + (P0[:,1]-p[...,1])*T[:,1]) / L\n", " U = U.reshape(len(U),1)\n", " D = P0 + U*T - p\n", " return np.sqrt((D**2).sum(axis=1))\n", "\n", "P0 = np.random.uniform(-10,10,(10,2))\n", "P1 = np.random.uniform(-10,10,(10,2))\n", "p = np.random.uniform(-10,10,( 1,2))\n", "print(distance(P0, P1, p))\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 4, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Numo::DFloat#shape=[10]\n", "[4.40292, 8.02008, 14.77, 16.1336, 16.0784, 7.29705, 9.79877, 10.7509, ...]\n" ] }, { "data": { "text/plain": [ "Numo::DFloat#shape=[10]\n", "[4.40292, 8.02008, 14.77, 16.1336, 16.0784, 7.29705, 9.79877, 10.7509, ...]" ] }, "execution_count": 4, "metadata": {}, "output_type": "execute_result" } ], "source": [ "def distance(p0, p1, p)\n", " t = p1 - p0\n", " l = (t**2).sum(1)\n", " u = -((p0[true,0]-p[false,0])*t[true,0] + (p0[true,1]-p[false,1])*t[true,1]) / l\n", " u = u.reshape(u.size,1)\n", " d = p0 + u*t - p\n", " return Numo::NMath.sqrt((d**2).sum(1))\n", "end\n", "\n", "p0 = Numo::DFloat.new(10,2).rand(-10,10)\n", "p1 = Numo::DFloat.new(10,2).rand(-10,10)\n", "p = Numo::DFloat.new( 1,2).rand(-10,10)\n", "p distance(p0, p1, p)" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 79. Consider 2 sets of points P0,P1 describing lines (2d) and a set of points P, how to compute distance from each point j (P\\[j\\]) to each line i (P0\\[i\\],P1\\[i\\])? (★★★)\n", "\n", "\n", "Python:\n", "```python\n", "# Author: Italmassov Kuanysh\n", "\n", "# based on distance function from previous question\n", "P0 = np.random.uniform(-10, 10, (10,2))\n", "P1 = np.random.uniform(-10,10,(10,2))\n", "p = np.random.uniform(-10, 10, (10,2))\n", "print(np.array([distance(P0,P1,p_i) for p_i in p]))\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 5, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "[Numo::DFloat#shape=[10]\n", "[4.02299, 3.7767, 7.22074, 2.65235, 4.03999, 0.0493129, 5.89712, 0.821607, ...], Numo::DFloat#shape=[10]\n", "[4.06287, 10.2757, 5.22386, 9.17308, 3.49263, 6.51912, 12.1396, 6.87145, ...], Numo::DFloat#shape=[10]\n", "[12.2731, 4.95469, 14.9322, 6.10746, 12.3248, 8.71038, 4.51429, 8.38228, ...], Numo::DFloat#shape=[10]\n", "[7.29208, 11.2986, 5.80238, 9.93285, 11.5081, 9.4783, 9.65492, 10.33, ...], Numo::DFloat#shape=[10]\n", "[7.42679, 14.6827, 6.16027, 13.4397, 8.20325, 10.8814, 16.0162, 11.3742, ...], Numo::DFloat#shape=[10]\n", "[8.90535, 14.3727, 6.22079, 12.941, 13.1026, 11.9889, 13.0645, 12.8597, ...], Numo::DFloat#shape=[10]\n", "[10.121, 14.4822, 11.257, 13.8115, 7.24394, 11.2893, 17.1786, 11.2277, ...], Numo::DFloat#shape=[10]\n", "[9.10298, 2.80676, 12.3148, 3.95976, 7.48986, 5.13894, 6.17195, 4.37329, ...], Numo::DFloat#shape=[10]\n", "[0.720696, 7.46437, 3.14777, 6.04, 4.88395, 4.03443, 8.16959, 4.84799, ...], Numo::DFloat#shape=[10]\n", "[2.82967, 10.3413, 0.872794, 8.92055, 6.33658, 6.80721, 10.889, 7.56309, ...]]\n" ] }, { "data": { "text/plain": [ "[Numo::DFloat#shape=[10]\n", "[4.02299, 3.7767, 7.22074, 2.65235, 4.03999, 0.0493129, 5.89712, 0.821607, ...], Numo::DFloat#shape=[10]\n", "[4.06287, 10.2757, 5.22386, 9.17308, 3.49263, 6.51912, 12.1396, 6.87145, ...], Numo::DFloat#shape=[10]\n", "[12.2731, 4.95469, 14.9322, 6.10746, 12.3248, 8.71038, 4.51429, 8.38228, ...], Numo::DFloat#shape=[10]\n", "[7.29208, 11.2986, 5.80238, 9.93285, 11.5081, 9.4783, 9.65492, 10.33, ...], Numo::DFloat#shape=[10]\n", "[7.42679, 14.6827, 6.16027, 13.4397, 8.20325, 10.8814, 16.0162, 11.3742, ...], Numo::DFloat#shape=[10]\n", "[8.90535, 14.3727, 6.22079, 12.941, 13.1026, 11.9889, 13.0645, 12.8597, ...], Numo::DFloat#shape=[10]\n", "[10.121, 14.4822, 11.257, 13.8115, 7.24394, 11.2893, 17.1786, 11.2277, ...], Numo::DFloat#shape=[10]\n", "[9.10298, 2.80676, 12.3148, 3.95976, 7.48986, 5.13894, 6.17195, 4.37329, ...], Numo::DFloat#shape=[10]\n", "[0.720696, 7.46437, 3.14777, 6.04, 4.88395, 4.03443, 8.16959, 4.84799, ...], Numo::DFloat#shape=[10]\n", "[2.82967, 10.3413, 0.872794, 8.92055, 6.33658, 6.80721, 10.889, 7.56309, ...]]" ] }, "execution_count": 5, "metadata": {}, "output_type": "execute_result" } ], "source": [ "p0 = Numo::DFloat.new(10,2).rand(-10,10)\n", "p1 = Numo::DFloat.new(10,2).rand(-10,10)\n", "p = Numo::DFloat.new(10,2).rand(-10,10)\n", "p a = p.shape[0].times.map{|i| distance(p0, p1, p.slice(i,true))}\n", "# todo: concat narray" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 80. Consider an arbitrary array, write a function that extract a subpart with a fixed shape and centered on a given element (pad with a `fill` value when necessary) (★★★)\n", "\n", "\n", "Python:\n", "```python\n", "# Author: Nicolas Rougier\n", "\n", "Z = np.random.randint(0,10,(10,10))\n", "shape = (5,5)\n", "fill = 0\n", "position = (1,1)\n", "\n", "R = np.ones(shape, dtype=Z.dtype)*fill\n", "P = np.array(list(position)).astype(int)\n", "Rs = np.array(list(R.shape)).astype(int)\n", "Zs = np.array(list(Z.shape)).astype(int)\n", "\n", "R_start = np.zeros((len(shape),)).astype(int)\n", "R_stop = np.array(list(shape)).astype(int)\n", "Z_start = (P-Rs//2)\n", "Z_stop = (P+Rs//2)+Rs%2\n", "\n", "R_start = (R_start - np.minimum(Z_start,0)).tolist()\n", "Z_start = (np.maximum(Z_start,0)).tolist()\n", "R_stop = np.maximum(R_start, (R_stop - np.maximum(Z_stop-Zs,0))).tolist()\n", "Z_stop = (np.minimum(Z_stop,Zs)).tolist()\n", "\n", "r = [slice(start,stop) for start,stop in zip(R_start,R_stop)]\n", "z = [slice(start,stop) for start,stop in zip(Z_start,Z_stop)]\n", "R[r] = Z[z]\n", "print(Z)\n", "print(R)\n", "```\n", "\n", "Ruby:\n", "```ruby\n", "# todo: minimum, maximum\n", "```" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 81. Consider an array Z = \\[1,2,3,4,5,6,7,8,9,10,11,12,13,14\\], how to generate an array R = \\[\\[1,2,3,4\\], \\[2,3,4,5\\], \\[3,4,5,6\\], ..., \\[11,12,13,14\\]\\]? (★★★)\n", "\n", "\n", "Python:\n", "```python\n", "# Author: Stefan van der Walt\n", "\n", "Z = np.arange(1,15,dtype=np.uint32)\n", "R = stride_tricks.as_strided(Z,(11,4),(4,4))\n", "print(R)\n", "```\n", "\n", "Ruby:\n", "```ruby\n", "# no moudle: stride_tricks\n", "```" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 82. Compute a matrix rank (★★★)\n", "\n", "\n", "Python:\n", "```python\n", "# Author: Stefan van der Walt\n", "\n", "Z = np.random.uniform(0,1,(10,10))\n", "U, S, V = np.linalg.svd(Z) # Singular Value Decomposition\n", "rank = np.sum(S > 1e-10)\n", "print(rank)\n", "```\n", "\n", "Ruby:\n", "```ruby\n", "# todo: svd\n", "```" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 83. How to find the most frequent value in an array?\n", "\n", "\n", "Python:\n", "```python\n", "Z = np.random.randint(0,10,50)\n", "print(np.bincount(Z).argmax())\n", "```\n", "\n", "Ruby:\n", "```ruby\n", "# todo: bincount\n", "```" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 84. Extract all the contiguous 3x3 blocks from a random 10x10 matrix (★★★)\n", "\n", "\n", "Python:\n", "```python\n", "# Author: Chris Barker\n", "\n", "Z = np.random.randint(0,5,(10,10))\n", "n = 3\n", "i = 1 + (Z.shape[0]-3)\n", "j = 1 + (Z.shape[1]-3)\n", "C = stride_tricks.as_strided(Z, shape=(i, j, n, n), strides=Z.strides + Z.strides)\n", "print(C)\n", "```\n", "\n", "Ruby:\n", "```ruby\n", "# no module: stride_tricks\n", "```" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 85. Create a 2D array subclass such that Z\\[i,j\\] == Z\\[j,i\\] (★★★)\n", "\n", "\n", "Python:\n", "```python\n", "# Author: Eric O. Lebigot\n", "# Note: only works for 2d array and value setting using indices\n", "\n", "class Symetric(np.ndarray):\n", " def __setitem__(self, index, value):\n", " i,j = index\n", " super(Symetric, self).__setitem__((i,j), value)\n", " super(Symetric, self).__setitem__((j,i), value)\n", "\n", "def symetric(Z):\n", " return np.asarray(Z + Z.T - np.diag(Z.diagonal())).view(Symetric)\n", "\n", "S = symetric(np.random.randint(0,10,(5,5)))\n", "S[2,3] = 42\n", "print(S)\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 6, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Numo::Int32#shape=[5,5]\n", "[[8, 6, 5, 8, 12], \n", " [6, 7, 9, 11, 10], \n", " [5, 9, 5, 42, 6], \n", " [8, 11, 42, 3, 16], \n", " [12, 10, 6, 16, 0]]\n" ] }, { "data": { "text/plain": [ "Numo::Int32#shape=[5,5]\n", "[[8, 6, 5, 8, 12], \n", " [6, 7, 9, 11, 10], \n", " [5, 9, 5, 42, 6], \n", " [8, 11, 42, 3, 16], \n", " [12, 10, 6, 16, 0]]" ] }, "execution_count": 6, "metadata": {}, "output_type": "execute_result" } ], "source": [ "module Symetric\n", " def []=(i,j,value)\n", " super(i,j,value)\n", " super(j,i,value) if i != j\n", " end\n", "end\n", "\n", "def symetric(z)\n", " y = z + z.transpose\n", " y.diagonal.store(z.diagonal)\n", " y.extend(Symetric)\n", "end\n", "\n", "s = symetric(Numo::Int32.new(5,5).rand(10))\n", "s[2,3] = 42\n", "p s" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 86. Consider a set of p matrices wich shape (n,n) and a set of p vectors with shape (n,1). How to compute the sum of of the p matrix products at once? (result has shape (n,1)) (★★★)\n", "\n", "\n", "Python:\n", "```python\n", "# Author: Stefan van der Walt\n", "\n", "p, n = 10, 20\n", "M = np.ones((p,n,n))\n", "V = np.ones((p,n,1))\n", "S = np.tensordot(M, V, axes=[[0, 2], [0, 1]])\n", "print(S)\n", "\n", "# It works, because:\n", "# M is (p,n,n)\n", "# V is (p,n,1)\n", "# Thus, summing over the paired axes 0 and 0 (of M and V independently),\n", "# and 2 and 1, to remain with a (n,1) vector.\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 7, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Numo::DFloat#shape=[20]\n", "[200, 200, 200, 200, 200, 200, 200, 200, 200, 200, 200, 200, 200, 200, 200, ...]\n" ] }, { "data": { "text/plain": [ "Numo::DFloat#shape=[20]\n", "[200, 200, 200, 200, 200, 200, 200, 200, 200, 200, 200, 200, 200, 200, 200, ...]" ] }, "execution_count": 7, "metadata": {}, "output_type": "execute_result" } ], "source": [ "p, n = 10, 20\n", "m = Numo::DFloat.ones(p,n,n)\n", "v = Numo::DFloat.ones(p,n,1)\n", "s = m.transpose(0,2,1).mulsum(v,0,1)\n", "p s\n", "# todo: tensordot?" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 87. Consider a 16x16 array, how to get the block-sum (block size is 4x4)? (★★★)\n", "\n", "\n", "Python:\n", "```python\n", "# Author: Robert Kern\n", "\n", "Z = np.ones((16,16))\n", "k = 4\n", "S = np.add.reduceat(np.add.reduceat(Z, np.arange(0, Z.shape[0], k), axis=0),\n", " np.arange(0, Z.shape[1], k), axis=1)\n", "print(S)\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 8, "metadata": { "collapsed": false }, "outputs": [ { "data": { "text/plain": [ "Numo::DFloat#shape=[4,4]\n", "[[16, 16, 16, 16], \n", " [16, 16, 16, 16], \n", " [16, 16, 16, 16], \n", " [16, 16, 16, 16]]" ] }, "execution_count": 8, "metadata": {}, "output_type": "execute_result" } ], "source": [ "n, k = 16, 4\n", "z = Numo::DFloat.ones(n,n)\n", "s = z.reshape(n/k,k,n/k,k).sum(1,3)\n", "# todo: reduceat?" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 88. How to implement the Game of Life using numpy arrays? (★★★)\n", "\n", "\n", "Python:\n", "```python\n", "# Author: Nicolas Rougier\n", "\n", "def iterate(Z):\n", " # Count neighbours\n", " N = (Z[0:-2,0:-2] + Z[0:-2,1:-1] + Z[0:-2,2:] +\n", " Z[1:-1,0:-2] + Z[1:-1,2:] +\n", " Z[2: ,0:-2] + Z[2: ,1:-1] + Z[2: ,2:])\n", "\n", " # Apply rules\n", " birth = (N==3) & (Z[1:-1,1:-1]==0)\n", " survive = ((N==2) | (N==3)) & (Z[1:-1,1:-1]==1)\n", " Z[...] = 0\n", " Z[1:-1,1:-1][birth | survive] = 1\n", " return Z\n", "\n", "Z = np.random.randint(0,2,(50,50))\n", "for i in range(100): Z = iterate(Z)\n", "print(Z)\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 9, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Numo::Int32#shape=[50,50]\n", "[[0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, ...], \n", " [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, ...], \n", " [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, ...], \n", " [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, ...], \n", " [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, ...], \n", " [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, ...], \n", " [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, ...], \n", " [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, ...], \n", " [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, ...], \n", " [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, ...], \n", " [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, ...], \n", " [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, ...], \n", " [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, ...], \n", " [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 1, 0, 0, 0, 0, 0, 0, 1, 1, 0, 0, ...], \n", " [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 1, 1, 0, 0, ...], \n", " [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, ...], \n", " [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 0, 1, 0, 1, 0, 0, 0, 0, 0, ...], \n", " [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 1, 0, 1, 0, 0, 0, 0, 0, ...], \n", " [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 0, 0, 0, 0, 1, 1, 0, 0, 0, 0, 0, 0, ...], \n", " [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, ...], \n", " ...\n" ] }, { "data": { "text/plain": [ "Numo::Int32#shape=[50,50]\n", "[[0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, ...], \n", " [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, ...], \n", " [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, ...], \n", " [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, ...], \n", " [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, ...], \n", " [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, ...], \n", " [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, ...], \n", " [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, ...], \n", " [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, ...], \n", " [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, ...], \n", " [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, ...], \n", " [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, ...], \n", " [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, ...], \n", " [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 1, 0, 0, 0, 0, 0, 0, 1, 1, 0, 0, ...], \n", " [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 1, 1, 0, 0, ...], \n", " [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, ...], \n", " [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 0, 1, 0, 1, 0, 0, 0, 0, 0, ...], \n", " [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 1, 0, 1, 0, 0, 0, 0, 0, ...], \n", " [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 0, 0, 0, 0, 1, 1, 0, 0, 0, 0, 0, 0, ...], \n", " [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, ...], \n", " ..." ] }, "execution_count": 9, "metadata": {}, "output_type": "execute_result" } ], "source": [ "def iterate(z)\n", " # Count neighbours\n", " n = z[0..-3,0..-3] + z[0..-3,1..-2] + z[0..-3,2..-1] +\n", " z[1..-2,0..-3] + z[1..-2,2..-1] +\n", " z[2..-1,0..-3] + z[2..-1,1..-2] + z[2..-1,2..-1]\n", "\n", " # Apply rules\n", " birth = n.eq(3) & z[1..-2,1..-2].eq(0)\n", " survive = (n.eq(2) | n.eq(3)) & z[1..-2,1..-2].eq(1)\n", " z[] = 0\n", " #z[1..-2,1..-2][birth | survive] = 1\n", " y = z[0..-3,0..-3].copy\n", " y[birth | survive] = 1\n", " z[1..-2,1..-2] = y\n", "end\n", "\n", "z = Numo::Int32.new(50,50).rand(2)\n", "100.times{ iterate(z) }\n", "p z" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 89. How to get the n largest values of an array (★★★)\n", "\n", "\n", "Python:\n", "```python\n", "Z = np.arange(10000)\n", "np.random.shuffle(Z)\n", "n = 5\n", "\n", "# Slow\n", "print (Z[np.argsort(Z)[-n:]])\n", "\n", "# Fast\n", "print (Z[np.argpartition(-Z,n)[:n]])\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 10, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Numo::DFloat(view)#shape=[5]\n", "[0.999617, 0.999781, 0.999866, 0.999873, 0.999884]\n" ] }, { "data": { "text/plain": [ "Numo::DFloat(view)#shape=[5]\n", "[0.999617, 0.999781, 0.999866, 0.999873, 0.999884]" ] }, "execution_count": 10, "metadata": {}, "output_type": "execute_result" } ], "source": [ "z = Numo::DFloat.new(10000).rand\n", "n = 5\n", "p z[z.sort_index[-n..-1]]\n", "# todo: shuffle, argpartition" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 90. Given an arbitrary number of vectors, build the cartesian product (every combinations of every item) (★★★)\n", "\n", "\n", "Python:\n", "```python\n", "# Author: Stefan Van der Walt\n", "\n", "def cartesian(arrays):\n", " arrays = [np.asarray(a) for a in arrays]\n", " shape = (len(x) for x in arrays)\n", "\n", " ix = np.indices(shape, dtype=int)\n", " ix = ix.reshape(len(arrays), -1).T\n", "\n", " for n, arr in enumerate(arrays):\n", " ix[:, n] = arrays[n][ix[:, n]]\n", "\n", " return ix\n", "\n", "print (cartesian(([1, 2, 3], [4, 5], [6, 7])))\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 11, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Numo::Int32#shape=[12,3]\n", "[[1, 4, 6], \n", " [1, 4, 7], \n", " [1, 5, 6], \n", " [1, 5, 7], \n", " [2, 4, 6], \n", " [2, 4, 7], \n", " [2, 5, 6], \n", " [2, 5, 7], \n", " [3, 4, 6], \n", " [3, 4, 7], \n", " [3, 5, 6], \n", " [3, 5, 7]]\n" ] }, { "data": { "text/plain": [ "Numo::Int32#shape=[12,3]\n", "[[1, 4, 6], \n", " [1, 4, 7], \n", " [1, 5, 6], \n", " [1, 5, 7], \n", " [2, 4, 6], \n", " [2, 4, 7], \n", " [2, 5, 6], \n", " [2, 5, 7], \n", " [3, 4, 6], \n", " [3, 4, 7], \n", " [3, 5, 6], \n", " [3, 5, 7]]" ] }, "execution_count": 11, "metadata": {}, "output_type": "execute_result" } ], "source": [ "def cartesian(*arrays)\n", " arrays = arrays.map{|a| Numo::Int32.cast(a)}\n", " shape = arrays.map{|x| x.size}\n", " asz = arrays.size\n", "\n", " ix = Numo::Int32.zeros(*shape, asz)\n", " arrays.each_with_index do |arr,n|\n", " s = [1]*asz\n", " s[n] = arr.size\n", " ix[false,n] = arr.reshape(*s)\n", " end\n", " return ix.reshape(ix.size/asz,asz)\n", "end\n", "\n", "p cartesian([1, 2, 3], [4, 5], [6, 7])" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 91. How to create a record array from a regular array? (★★★)\n", "\n", "\n", "Python:\n", "```python\n", "Z = np.array([(\"Hello\", 2.5, 3),\n", " (\"World\", 3.6, 2)])\n", "R = np.core.records.fromarrays(Z.T,\n", " names='col1, col2, col3',\n", " formats = 'S8, f8, i8')\n", "print(R)\n", "```\n", "\n", "Ruby:\n", "```ruby\n", "# todo: record\n", "```" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 92. Consider a large vector Z, compute Z to the power of 3 using 3 different methods (★★★)\n", "\n", "\n", "Python:\n", "```python\n", "# Author: Ryan G.\n", "\n", "x = np.random.rand(5e7)\n", "\n", "%timeit np.power(x,3)\n", "%timeit x*x*x\n", "%timeit np.einsum('i,i,i->i',x,x,x)\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 2, "metadata": { "collapsed": false }, "outputs": [ { "data": { "text/plain": [ "Numo::DFloat#shape=[50000000]\n", "[0.000235508, 0.051923, 0.50211, 0.00812571, 0.00156255, 0.040719, ...]" ] }, "execution_count": 2, "metadata": {}, "output_type": "execute_result" } ], "source": [ "x = Numo::DFloat.new(5e7).rand\n", "x**3 # probably fast" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 93. Consider two arrays A and B of shape (8,3) and (2,2). How to find rows of A that contain elements of each row of B regardless of the order of the elements in B? (★★★)\n", "\n", "\n", "Python:\n", "```python\n", "# Author: Gabe Schwartz\n", "\n", "A = np.random.randint(0,5,(8,3))\n", "B = np.random.randint(0,5,(2,2))\n", "\n", "C = (A[..., np.newaxis, np.newaxis] == B)\n", "rows = (C.sum(axis=(1,2,3)) >= B.shape[1]).nonzero()[0]\n", "print(rows)\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 3, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Numo::Int32#shape=[6]\n", "[0, 1, 2, 5, 6, 7]\n" ] }, { "data": { "text/plain": [ "Numo::Int32#shape=[6]\n", "[0, 1, 2, 5, 6, 7]" ] }, "execution_count": 3, "metadata": {}, "output_type": "execute_result" } ], "source": [ "a = Numo::Int32.new(8,3).rand(5)\n", "b = Numo::Int32.new(2,2).rand(5)\n", "c = a[false,:new,:new].eq b\n", "rows = (c.count_true(1,2,3) >= b.shape[1]).where\n", "p rows" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 94. Considering a 10x3 matrix, extract rows with unequal values (e.g. \\[2,2,3\\]) (★★★)\n", "\n", "\n", "Python:\n", "```python\n", "# Author: Robert Kern\n", "\n", "Z = np.random.randint(0,5,(10,3))\n", "E = np.logical_and.reduce(Z[:,1:] == Z[:,:-1], axis=1)\n", "U = Z[~E]\n", "print(Z)\n", "print(U)\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 4, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Numo::Int32#shape=[10,3]\n", "[[0, 2, 1], \n", " [3, 1, 0], \n", " [3, 1, 2], \n", " [2, 4, 4], \n", " [0, 3, 0], \n", " [4, 3, 3], \n", " [4, 3, 2], \n", " [2, 0, 2], \n", " [2, 3, 2], \n", " [2, 3, 4]]\n", "Numo::Int32(view)#shape=[10,3]\n", "[[0, 2, 1], \n", " [3, 1, 0], \n", " [3, 1, 2], \n", " [2, 4, 4], \n", " [0, 3, 0], \n", " [4, 3, 3], \n", " [4, 3, 2], \n", " [2, 0, 2], \n", " [2, 3, 2], \n", " [2, 3, 4]]\n" ] }, { "data": { "text/plain": [ "Numo::Int32(view)#shape=[10,3]\n", "[[0, 2, 1], \n", " [3, 1, 0], \n", " [3, 1, 2], \n", " [2, 4, 4], \n", " [0, 3, 0], \n", " [4, 3, 3], \n", " [4, 3, 2], \n", " [2, 0, 2], \n", " [2, 3, 2], \n", " [2, 3, 4]]" ] }, "execution_count": 4, "metadata": {}, "output_type": "execute_result" } ], "source": [ "z = Numo::Int32.new(10,3).rand(5)\n", "e = (z[true,1..-1].eq z[true,0..-2]).all?(1)\n", "u = z[(~e).where,true]\n", "p z\n", "p u" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 95. Convert a vector of ints into a matrix binary representation (★★★)\n", "\n", "\n", "Python:\n", "```python\n", "# Author: Warren Weckesser\n", "\n", "I = np.array([0, 1, 2, 3, 15, 16, 32, 64, 128])\n", "B = ((I.reshape(-1,1) & (2**np.arange(8))) != 0).astype(int)\n", "print(B[:,::-1])\n", "\n", "# Author: Daniel T. McDonald\n", "\n", "I = np.array([0, 1, 2, 3, 15, 16, 32, 64, 128], dtype=np.uint8)\n", "print(np.unpackbits(I[:, np.newaxis], axis=1))\n", "```\n", "\n", "Ruby:\n", "```ruby\n", "# todo: bit\n", "```" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 96. Given a two dimensional array, how to extract unique rows? (★★★)\n", "\n", "\n", "Python:\n", "```python\n", "# Author: Jaime Fernández del Río\n", "\n", "Z = np.random.randint(0,2,(6,3))\n", "T = np.ascontiguousarray(Z).view(np.dtype((np.void, Z.dtype.itemsize * Z.shape[1])))\n", "_, idx = np.unique(T, return_index=True)\n", "uZ = Z[idx]\n", "print(uZ)\n", "```\n", "\n", "Ruby:\n", "```ruby\n", "# todo: unique row\n", "```" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 97. Considering 2 vectors A & B, write the einsum equivalent of inner, outer, sum, and mul function (★★★)\n", "\n", "\n", "Python:\n", "```python\n", "# Author: Alex Riley\n", "# Make sure to read: http://ajcr.net/Basic-guide-to-einsum/\n", "\n", "A = np.random.uniform(0,1,10)\n", "B = np.random.uniform(0,1,10)\n", "\n", "np.einsum('i->', A) # np.sum(A)\n", "np.einsum('i,i->i', A, B) # A * B\n", "np.einsum('i,i', A, B) # np.inner(A, B)\n", "np.einsum('i,j', A, B) # np.outer(A, B)\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 5, "metadata": { "collapsed": false }, "outputs": [ { "data": { "text/plain": [ "Numo::DFloat#shape=[10,10]\n", "[[0.418361, 0.252631, 0.168549, 0.367414, 0.0409535, 0.365022, 0.22084, ...], \n", " [0.369513, 0.223134, 0.148869, 0.324515, 0.0361717, 0.322402, 0.195055, ...], \n", " [0.575741, 0.347666, 0.231954, 0.505629, 0.0563594, 0.502337, 0.303916, ...], \n", " [0.423199, 0.255553, 0.170498, 0.371663, 0.0414271, 0.369243, 0.223394, ...], \n", " [0.0554262, 0.0334696, 0.02233, 0.0486766, 0.00542569, 0.0483596, ...], \n", " [0.285368, 0.172322, 0.114968, 0.250617, 0.0279347, 0.248985, 0.150637, ...], \n", " [0.398275, 0.240502, 0.160457, 0.349775, 0.0389873, 0.347497, 0.210238, ...], \n", " [0.21605, 0.130464, 0.0870418, 0.18974, 0.0211492, 0.188505, 0.114046, ...], \n", " [0.343856, 0.20764, 0.138532, 0.301982, 0.0336601, 0.300016, 0.181511, ...], \n", " [0.0467036, 0.0282024, 0.0188159, 0.0410162, 0.00457183, 0.0407492, ...]]" ] }, "execution_count": 5, "metadata": {}, "output_type": "execute_result" } ], "source": [ "# no method: einsum\n", "a = Numo::DFloat.new(10).rand(0,1)\n", "b = Numo::DFloat.new(10).rand(0,1)\n", "\n", "a.sum # np.sum(A)\n", "a*b # A * B\n", "a.mulsum(b) # np.inner(A, B)\n", "a[false,:new]*b # np.outer(A, B)" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 98. Considering a path described by two vectors (X,Y), how to sample it using equidistant samples (★★★)?\n", "\n", "\n", "Python:\n", "```python\n", "# Author: Bas Swinckels\n", "\n", "phi = np.arange(0, 10*np.pi, 0.1)\n", "a = 1\n", "x = a*phi*np.cos(phi)\n", "y = a*phi*np.sin(phi)\n", "\n", "dr = (np.diff(x)**2 + np.diff(y)**2)**.5 # segment lengths\n", "r = np.zeros_like(x)\n", "r[1:] = np.cumsum(dr) # integrate path\n", "r_int = np.linspace(0, r.max(), 200) # regular spaced path\n", "x_int = np.interp(r_int, r, x) # integrate path\n", "y_int = np.interp(r_int, r, y)\n", "```\n", "\n", "Ruby:\n", "```ruby\n", "# todo: interp\n", "```" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "#### 99. Given an integer n and a 2D array X, select from X the rows which can be interpreted as draws from a multinomial distribution with n degrees, i.e., the rows which only contain integers and which sum to n. (★★★)\n", "\n", "\n", "Python:\n", "```python\n", "# Author: Evgeni Burovski\n", "\n", "X = np.asarray([[1.0, 0.0, 3.0, 8.0],\n", " [2.0, 0.0, 1.0, 1.0],\n", " [1.5, 2.5, 1.0, 0.0]])\n", "n = 4\n", "M = np.logical_and.reduce(np.mod(X, 1) == 0, axis=-1)\n", "M &= (X.sum(axis=-1) == n)\n", "print(X[M])\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "markdown", "metadata": { "collapsed": true }, "source": [ "#### 100. Compute bootstrapped 95% confidence intervals for the mean of a 1D array X (i.e., resample the elements of an array with replacement N times, compute the mean of each sample, and then compute percentiles over the means). (★★★)\n", "\n", "\n", "Python:\n", "```python\n", "# Author: Jessica B. Hamrick\n", "\n", "X = np.random.randn(100) # random 1D array\n", "N = 1000 # number of bootstrap samples\n", "idx = np.random.randint(0, X.size, (N, X.size))\n", "means = X[idx].mean(1)\n", "confint = np.percentile(means, [2.5, 97.5])\n", "print(confint)\n", "```\n", "\n", "Ruby:" ] }, { "cell_type": "code", "execution_count": 2, "metadata": { "collapsed": false }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Numo::DFloat(view)#shape=[2]\n", "[0.345385, 0.449567]\n" ] }, { "data": { "text/plain": [ "Numo::DFloat(view)#shape=[2]\n", "[0.345385, 0.449567]" ] }, "execution_count": 2, "metadata": {}, "output_type": "execute_result" } ], "source": [ "x = Numo::DFloat.new(100).rand\n", "n = 1000 # number of bootstrap samples\n", "idx = Numo::Int32.new(n, x.size).rand(x.size)\n", "means = x[idx].mean(1)\n", "confint = means[means.sort_index[means.size/100.0*Numo::DFloat[2.5, 97.5]]]\n", "p confint\n", "# todo: percentile, rand_norm" ] }, { "cell_type": "code", "execution_count": null, "metadata": { "collapsed": true }, "outputs": [], "source": [] } ], "metadata": { "kernelspec": { "display_name": "Ruby 2.3.1", "language": "ruby", "name": "ruby" }, "language_info": { "file_extension": ".rb", "mimetype": "application/x-ruby", "name": "ruby", "version": "2.3.1" } }, "nbformat": 4, "nbformat_minor": 0 }