예제 #1
0
파일: prep.py 프로젝트: sbeef/dirty-scripts
def plot(p1, p2, name):
    gsamps = samples.load('gsa_samples/GSA_Samples.data')
    good_samps = [sample for sample in gsamps if sample.pb is not None]
    pb_vals = [fix(sample.pb) for sample in good_samps]
    cs_vals = [sample.cs for sample in good_samps]
    cult_vals = [sample.cultivated for sample in good_samps]
    art_vals = [sample.artificial for sample in good_samps]
    rain_vals = [sample.rain.mean for sample in good_samps]
    relief_vals = [sample.relief.mean for sample in good_samps]
    elevation_vals = [sample.elevation.mean for sample in good_samps]
    slope_vals = [sample.slope.mean for sample in good_samps]
    area_vals = [sample.area/1000000 for sample in good_samps]
    pb = (pb_vals, LABELS['pb'], 'pb')
    cs = (cs_vals, LABELS['cs'], 'cs')
    cult = (cult_vals, LABELS['cult'], 'cult')
    art = (art_vals, LABELS['art'], 'art')
    rain = (rain_vals, LABELS['rain'], 'rain')
    relief = (relief_vals, LABELS['relief'], 'relief')
    elevation = (elevation_vals, LABELS['elevation'], 'elev')
    slope = (slope_vals, LABELS['slope'], 'slope')
    area = (area_vals, LABELS['area'], 'area')
    mapper = {
        'pb':pb,
        'cs':cs,
        'cult':cult,
        'cultivated':cult,
        'cultivation':cult,
        'art':art,
        'artificial':art,
        'rain':rain,
        'precip':rain,
        'precipitation':rain,
        'relief':relief,
        'elevation':elevation,
        'elev':elevation,
        'slope':slope,
        'area':area,
    }
    oname = "gsa_samples/plots/%s" % name
    samples.plot(mapper[p1], mapper[p2], oname)
예제 #2
0
파일: prep.py 프로젝트: sbeef/dirty-scripts
def gogographit(neg='nothing'):
    gsamps = samples.load('gsa_samples/GSA_Samples.data')
    good_samps = [sample for sample in gsamps if sample.pb is not None]
    if neg == 'remove':
        good_samps = [sample for sample in good_samps if sample.pb >= 0]
    if neg == 'zero':
        pb_vals = [fix(sample.pb) for sample in good_samps]
    else:
        pb_vals = [sample.pb for sample in good_samps]
    cs_vals = [sample.cs for sample in good_samps]
    cult_vals = [sample.cultivated for sample in good_samps]
    art_vals = [sample.artificial for sample in good_samps]
    rain_vals = [sample.rain.mean for sample in good_samps]
    relief_vals = [sample.relief.mean for sample in good_samps]
    elevation_vals = [sample.elevation.mean for sample in good_samps]
    slope_vals = [sample.slope.mean for sample in good_samps]
    area_vals = [sample.area/1000000 for sample in good_samps]
    pb = (pb_vals, LABELS['pb'], 'pb')
    cs = (cs_vals, LABELS['cs'], 'cs')
    cult = (cult_vals, LABELS['cult'], 'cult')
    art = (art_vals, LABELS['art'], 'art')
    rain = (rain_vals, LABELS['rain'], 'rain')
    relief = (relief_vals, LABELS['relief'], 'relief')
    elevation = (elevation_vals, LABELS['elevation'], 'elev')
    slope = (slope_vals, LABELS['slope'], 'slope')
    area = (area_vals, LABELS['area'], 'area')
    indvars = (cult, art, rain, relief, elevation, slope, area)
    plot_folder = 'gsa_samples/plots'
    if neg == 'zero':
        extra = "_zero"
    elif neg == 'remove':
        extra = "no-neg"
    else:
        extra = ""
    for var in indvars:
        print "plotting %s against %s" % (var[2], 'pb')
        fname = "%s/%s_pb%s" % (plot_folder, var[2], extra)
        samples.plot(var, pb, fname)