masked_weirdness.ipynb

{
 "metadata": {
  "name": ""
 },
 "nbformat": 3,
 "nbformat_minor": 0,
 "worksheets": [
  {
   "cells": [
    {
     "cell_type": "code",
     "collapsed": false,
     "input": [
      "import numpy as np"
     ],
     "language": "python",
     "metadata": {},
     "outputs": [],
     "prompt_number": 1
    },
    {
     "cell_type": "code",
     "collapsed": false,
     "input": [
      "testarr =np.array([[[1,2,3,4],[5,6,7,8],[9,10,11,12]],[[1,2,3,4],[5,6,7,8],[9,10,11,12]]])\n",
      "print testarr\n",
      "print testarr<9\n",
      "volarr = np.ones(testarr.shape)\n",
      "print volarr[testarr<9].sum()\n",
      "maskarr =np.array([[[1,0,0,1],[0,1,1,0],[0,1,1,0]],[[1,0,0,1],[0,1,1,0],[0,1,1,0]]])\n",
      "testarr = np.ma.masked_where(maskarr==0,testarr)\n",
      "#volarr = np.ma.masked_where(maskarr==0,volarr)\n",
      "print testarr\n",
      "print testarr<9\n",
      "print np.ma.sum(volarr[testarr<9])"
     ],
     "language": "python",
     "metadata": {},
     "outputs": [
      {
       "output_type": "stream",
       "stream": "stdout",
       "text": [
        "[[[ 1  2  3  4]\n",
        "  [ 5  6  7  8]\n",
        "  [ 9 10 11 12]]\n",
        "\n",
        " [[ 1  2  3  4]\n",
        "  [ 5  6  7  8]\n",
        "  [ 9 10 11 12]]]\n",
        "[[[ True  True  True  True]\n",
        "  [ True  True  True  True]\n",
        "  [False False False False]]\n",
        "\n",
        " [[ True  True  True  True]\n",
        "  [ True  True  True  True]\n",
        "  [False False False False]]]\n",
        "16.0\n",
        "[[[1 -- -- 4]\n",
        "  [-- 6 7 --]\n",
        "  [-- 10 11 --]]\n",
        "\n",
        " [[1 -- -- 4]\n",
        "  [-- 6 7 --]\n",
        "  [-- 10 11 --]]]\n",
        "[[[True -- -- True]\n",
        "  [-- True True --]\n",
        "  [-- False False --]]\n",
        "\n",
        " [[True -- -- True]\n",
        "  [-- True True --]\n",
        "  [-- False False --]]]\n",
        "16.0\n"
       ]
      }
     ],
     "prompt_number": 2
    },
    {
     "cell_type": "code",
     "collapsed": false,
     "input": [
      "print ~testarr.mask"
     ],
     "language": "python",
     "metadata": {},
     "outputs": [
      {
       "output_type": "stream",
       "stream": "stdout",
       "text": [
        "[[[ True False False  True]\n",
        "  [False  True  True False]\n",
        "  [False  True  True False]]\n",
        "\n",
        " [[ True False False  True]\n",
        "  [False  True  True False]\n",
        "  [False  True  True False]]]\n"
       ]
      }
     ],
     "prompt_number": 4
    },
    {
     "cell_type": "code",
     "collapsed": false,
     "input": [
      "print testarr.shape"
     ],
     "language": "python",
     "metadata": {},
     "outputs": [
      {
       "output_type": "stream",
       "stream": "stdout",
       "text": [
        "(2, 3, 4)\n"
       ]
      }
     ],
     "prompt_number": 5
    },
    {
     "cell_type": "code",
     "collapsed": false,
     "input": [
      "print testarr[1,:,:]\n",
      "print testarr.mask[1,:,:]"
     ],
     "language": "python",
     "metadata": {},
     "outputs": [
      {
       "output_type": "stream",
       "stream": "stdout",
       "text": [
        "[[1 -- -- 4]\n",
        " [-- 6 7 --]\n",
        " [-- 10 11 --]]\n",
        "[[False  True  True False]\n",
        " [ True False False  True]\n",
        " [ True False False  True]]\n"
       ]
      }
     ],
     "prompt_number": 7
    },
    {
     "cell_type": "code",
     "collapsed": false,
     "input": [],
     "language": "python",
     "metadata": {},
     "outputs": []
    }
   ],
   "metadata": {}
  }
 ]
}