forked from rlowrance/re-avm
-
Notifications
You must be signed in to change notification settings - Fork 0
/
transactions.py
324 lines (265 loc) · 9.38 KB
/
transactions.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
'''join deeds and tax roll files to create arms-length, grant, and sfr transactions
INPUT FILES
INPUT/real-estate-log-angeles/corelogic-deeds-090402_07/CAC06037F1.zip ...
INPUT/real-estate-log-angeles/corelogic-deeds-090402_09/CAC06037F1.zip ...
INPUT/real-estate-log-angeles/corelogic-taxrolls-090402_05/CAC06037F1.zip ...
WORKING/parcels-features-census_tract.csv
WORKING/parcels-features-zip5.csv
OUTPUT FILES
WORKING/transactions-al-g-sfr.csv
NOTES:
1. The deeds file has the prior sale info, which we could use
to create more transactions. We didn't, because we only have
census data from the year 2000, and we use census tract
features, so its not effective to go back before sometime
in 2002, when the 2000 census data became public.
2. Fields created in this program
avg_commute
fraction_owner_occupied
median_household_income
X_has_commercial
X_has_industry
X_has_park
X_has_retail
X_has_school, for X in {census_tract, zip5}
best_apn
zip5
'''
import numpy as np
import pandas as pd
import pdb
from pprint import pprint
import random
import sys
from Bunch import Bunch
from columns_contain import columns_contain
import layout_deeds as deeds
import layout_parcels as parcels
import layout_transactions as transactions
from Logger import Logger
import pandas_utilities as pu
from ParseCommandLine import ParseCommandLine
from Path import Path
cc = columns_contain
def usage(msg=None):
if msg is not None:
print msg
print 'usage : python transactions.py [--test]'
print ' --test: run in test mode'
sys.exit(1)
def make_control(argv):
# return a Bunch
print argv
if len(argv) not in (1, 2):
usage('invalid number of arguments')
pcl = ParseCommandLine(argv)
if pcl.has_arg('--help'):
usage()
arg = Bunch(
base_name=argv[0].split('.')[0],
test=pcl.has_arg('--test'),
)
random_seed = 123456
random.seed(random_seed)
path = Path() # use the default dir_input
debug = False
file_out_transactions = (
('testing-' if arg.test else '') +
arg.base_name + '-al-g-sfr' + '.csv'
)
return Bunch(
arg=arg,
debug=debug,
max_sale_price=85e6, # according to Wall Street Journal
path=path,
path_in_census_features=path.dir_working() + 'census-features-derived.csv',
path_in_parcels_features_census_tract=path.dir_working() + 'parcels-features-census_tract.csv',
path_in_parcels_features_zip5=path.dir_working() + 'parcels-features-zip5.csv',
path_out_transactions=path.dir_working() + file_out_transactions,
random_seed=random_seed,
test=arg.test,
)
def features(name, df):
'return features that contain name'
return [x for x in df.columns if name.lower() in x.lower()]
def best_apn(df, feature_formatted, feature_unformatted):
'''return series with best apn
Algo for the R version
use unformatted, if its all digits
otherwise, use formatted, if removing hyphens makes a number
otherwise, use NaN
'''
formatted = df[feature_formatted]
unformatted = df[feature_unformatted]
if False:
print unformatted.head()
print formatted.head()
if np.dtype(unformatted) == np.int64:
# every entry is np.int64, because pd.read_csv made it so
return unformatted
if np.dtype(unformatted) == np.object:
return np.int64(unformatted)
print 'not expected'
pdb.set_trace()
def read_census_features(control):
'return dataframe'
print 'reading census'
df = pd.read_csv(control.path_in_census_features, index_col=0)
return df
def read_geocoding(control):
'return dataframe'
print 'reading geocoding'
df = pd.read_csv(control.path.dir_input('geocoding'), sep='\t')
return df
def parcels_derived_features(control, transactions_df):
'return new df by merging df and the geo features'
# merge in census tract features
census_tract_df = pd.read_csv(control.path_in_parcels_features_census_tract, index_col=0)
check_feature_names(transactions_df)
check_feature_names(census_tract_df)
m1 = transactions_df.merge(
census_tract_df,
how='inner',
left_on=transactions_df[transactions.census_tract],
right_on=census_tract_df.census_tract,
)
check_feature_names(m1)
print 'm1 shape', m1.shape
cc('commercial', m1)
# merge in zip5 features
zip5_df = pd.read_csv(control.path_in_parcels_features_zip5, index_col=0)
check_feature_names(m1)
check_feature_names(zip5_df)
m2 = m1.merge(
zip5_df,
how='inner',
left_on=m1[transactions.zip5],
right_on=zip5_df.zip5,
)
# remove duplicated field zip5_x, zip5_y
assert 'zip5_x' in m2.columns
assert 'zip5_y' in m2.columns
assert (m2.zip5_x == m2.zip5_y).all()
assert 'zip5' not in m2.columns
pu.df_remove_column(m2, 'zip5_x')
pu.df_rename_column(m2, 'zip5_y', 'zip5')
check_feature_names(m2)
print 'm2 shape', m2.shape
return m2
def check_feature_names(df):
'stop if an unexpected feature name appears'
# was used for debugging, but left in the production version
# in case this type of checking needs to be done again later
for name in df.columns:
if name.endswith('_x') or name.endswith('_y'):
print 'check_feature_names; bad name:', name
pdb.set_trace()
def main(argv):
control = make_control(argv)
sys.stdout = Logger(base_name=control.arg.base_name)
print control
# NOTE: Organize the computation to minimize memory usage
# so that this code can run on smaller-memory systems
def ps(name, value):
s = value.shape
print ' %20s shape (%d, %d)' % (name, s[0], s[1])
# create dataframes
n_read_if_test = 10000
deeds_g_al = deeds.read_g_al(
control.path,
n_read_if_test if control.test else None,
)
parcels_sfr = parcels.read(
control.path,
10000 if control.test else None,
just_sfr=True,
)
ps('original deeds g al', deeds_g_al)
ps('original parcels sfr', parcels_sfr)
# augment parcels to include a zip5 field (5-digit zip code)
# drop samples without a zipcode
# rationale: we use the zip5 to join the features derived from parcels
# and zip5 is derived from zipcode
zipcode_present = parcels_sfr[parcels.zipcode].notnull()
parcels_sfr = parcels_sfr[zipcode_present]
parcels.add_zip5(parcels_sfr)
# augment parcels and deeds to include a better APN
print 'adding best apn column for parcels'
new_column_parcels = best_apn(parcels_sfr, parcels.apn_formatted, parcels.apn_unformatted)
parcels_sfr.loc[:, parcels.best_apn] = new_column_parcels # generates an ignorable warning
print 'adding best apn column for deeds'
new_column_deeds = best_apn(deeds_g_al, deeds.apn_formatted, deeds.apn_unformatted)
deeds_g_al.loc[:, deeds.best_apn] = new_column_deeds
ps('revised deeds_g_al', deeds_g_al)
ps('revised parcels_sfr', parcels_sfr)
# join the deeds and parcels files
print 'starting to merge'
check_feature_names(deeds_g_al)
check_feature_names(parcels_sfr)
m1 = deeds_g_al.merge(parcels_sfr, how='inner',
left_on=deeds.best_apn, right_on=parcels.best_apn,
suffixes=('_deed', '_parcel'))
check_feature_names(m1)
del deeds_g_al
del parcels_sfr
ps('m1 merge deed + parcels', m1)
# add in derived parcels features
m2 = parcels_derived_features(control, m1)
check_feature_names(m2)
ps('ms added parcels_derived', m2)
del m1
# add in census data
census_features_df = read_census_features(control)
m3 = m2.merge(census_features_df,
left_on=transactions.census_tract,
right_on="census_tract",
)
assert 'census_tract_x' in m3.columns
assert 'census_tract_y' in m3.columns
assert (m3.census_tract_x == m3.census_tract_y).all()
assert 'census_tract' not in m3.columns
pu.df_remove_column(m3, 'census_tract_x')
pu.df_rename_column(m3, 'census_tract_y', 'census_tract')
check_feature_names(m3)
del m2
ps('m3 merged census features', m3)
# add in GPS coordinates
geocoding_df = read_geocoding(control)
m4 = m3.merge(geocoding_df,
left_on="best_apn",
right_on="G APN",
)
del geocoding_df
del m3
ps('m4 merged geocoding', m4)
final = m4
print 'final columns'
for c in final.columns:
print c,
print
cc('fraction', final) # verify that fraction_owner_occupied is in the output
print 'final shape', final.shape
# write merged,augmented dataframe
print 'writing final dataframe to csv file'
final.to_csv(control.path_out_transactions)
# write out all the column names
print 'all column names in final dataframe'
for name in final.columns:
print name
if '_y' in name:
print 'found strange suffix'
pdb.set_trace()
print control
if control.test:
print 'DISCARD OUTPUT: test'
print 'done'
return
if __name__ == '__main__':
if False:
# avoid pyflakes warnings
pdb.set_trace()
pprint()
pd.DataFrame()
np.array()
columns_contain('s', pd.DataFrame())
main(sys.argv)