Skip to content

Instantly share code, notes, and snippets.

@henryroe
henryroe / import_nsf_ast_csv.py
Created June 20, 2014 21:37
Import NSF AST funding data in CSV format
import pandas
import numpy as np
awards = pandas.read_csv("awards_dump_2014-06-20.csv",
parse_dates=['StartDate', 'LastAmendmentDate', 'ExpirationDate',
'AwardedAmountToDate'],
converters={'AwardedAmountToDate': lambda x:
float(x.replace('$', '').replace(',', ''))},
dtype={'AwardedAmountToDate':np.float64})
awards['DurationYears'] = ((awards['ExpirationDate'] - awards['StartDate']) /
(365.25 * np.timedelta64(1, 'D')))
@henryroe
henryroe / plot_nsf_ast_grants_per_year.py
Created June 20, 2014 21:40
Plot NSF AST funding data for per-year funding of grants
max_award_per_year = 3e5 # above this amount just bin in to top bin
years_per_bin = 1.0
award_bin_size = 10000
start_year = np.array([a.year + (a.month - 1)/12. for a in awards['StartDate']])
year_bins = np.arange(np.floor(start_year.min()), np.ceil(start_year.max())+1, years_per_bin)
awarded_per_year = np.array(awards['AwardedAmountToDate']/awards['DurationYears'])
awarded_per_year_mean_per_bin = np.zeros(year_bins.size - 1)
awarded_per_year_median_per_bin = np.zeros(year_bins.size - 1)
year_per_bin = np.zeros(year_bins.size - 1)
for i in np.arange(year_bins.size - 1):
@henryroe
henryroe / plot_nsf_ast_grant_totals.py
Created June 20, 2014 21:41
Plot NSF AST funding data for total funding of grants
max_award = 1e6 # above this amount just bin in to top bin
years_per_bin = 1.0
award_bin_size = 50000
start_year = np.array([a.year + (a.month - 1)/12. for a in awards['StartDate']])
year_bins = np.arange(np.floor(start_year.min()), np.ceil(start_year.max())+1, years_per_bin)
awarded = np.array(awards['AwardedAmountToDate'])
awarded_mean_per_bin = np.zeros(year_bins.size - 1)
awarded_median_per_bin = np.zeros(year_bins.size - 1)
year_per_bin = np.zeros(year_bins.size - 1)
for i in np.arange(year_bins.size - 1):
@henryroe
henryroe / talk2subprocess_traitsui.py
Created August 25, 2014 22:40
Demo of launching a traitsui GUI in a separate thread and communicating with it via stdin and stdout
import time
import os
import sys
import pickle
import numpy as np
import subprocess
from threading import Thread
from traits.api import HasTraits, String, Instance, Bool, on_trait_change, Long
from traitsui.api import View, Item, Handler
import psutil
@henryroe
henryroe / bad_idea.py
Created October 3, 2014 00:48
Example of why you shouldn't use a mutable as a default parameter in python
class BadIdea():
def __init__(self, history=[]):
self.history = history
def get_history(self):
return self.history
def append_history(self, input):
self.history.append(input)
@henryroe
henryroe / prepend_and_save.scpt
Created February 14, 2015 12:03
Save mail attachments after pre-pending YYYY-MM-DD_
set outputPath to choose folder
tell application "Mail"
set curMessage to selection
set listAttachments to mail attachment of item 1 of curMessage
set curMessageDate to date received of item 1 of curMessage
set dateStr to (rich text -4 thru -1 of ("0000" & (year of curMessageDate))) & "-" & ¬
(rich text -2 thru -1 of ("00" & ((month of curMessageDate) as integer))) & "-" & ¬
(rich text -2 thru -1 of ("00" & (day of curMessageDate)))
repeat with a from 1 to length of listAttachments
@henryroe
henryroe / scene_scroll_test.py
Created September 27, 2013 04:24
scene_scroll_test
from scene import *
from math import exp
from threading import Thread
import datetime
# example scrolling scene with inertial scrolling
# basic scrolling example was by Dalorbi on the forums at:
# http://omz-software.com/pythonista/forums/discussion/213/scrolling-in-scene-module/p1
# inertial scrolling added on by hroe
@henryroe
henryroe / Rotate portrait pages by 90
Created January 6, 2014 22:20
Rotate portrait pages by +90degrees in current document in PDFpenPro 6
# Install this file in:
# ~/Library/Application Scripts/com.smileonmymac.PDFpenPro6.MacAppStore/
# and it will then be accessible from the Applescript menu in PDFpenPro 6
tell application "PDFpenPro 6"
if (count documents) > 0 then
set myDoc to document 1
set pageCount to count pages of myDoc
repeat with pageNumber from 1 to pageCount
if (height of page pageNumber of myDoc) > (width of page pageNumber of myDoc) then
@henryroe
henryroe / lessfits
Created July 26, 2016 03:55
lessfits: Command line script for paging through a FITS file's headers
#!/bin/bash
# This script is released under an "MIT License"; see https://opensource.org/licenses/MIT
# The MIT License (MIT)
# Copyright (c) 2016 Henry Roe (hroe@hroe.me)
#
# Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.
#
@henryroe
henryroe / grepfits
Last active July 26, 2016 04:11
grepfits: Command line script for summarizing selected keywords from headers of one or more FITS files
#!/usr/bin/env python
# This script is released under an "MIT License"; see https://opensource.org/licenses/MIT
# The MIT License (MIT)
# Copyright (c) 2016 Henry Roe (hroe@hroe.me)
#
# Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.
#