36 lines
814 B
Python
36 lines
814 B
Python
|
import rasterio
|
||
|
from constants import *
|
||
|
import matplotlib.pyplot as plt
|
||
|
from map_generator import predict_end_to_end
|
||
|
import pandas as pd
|
||
|
|
||
|
MARS=GEODATA / 'planets' / 'Mars_MGS_MOLA_DEM_mosaic_global_463m.tif'
|
||
|
elevation = rasterio.open(ELEVATION)
|
||
|
elevation_data = elevation.read(1)
|
||
|
|
||
|
|
||
|
data = {
|
||
|
'longitude': [],
|
||
|
'latitude': [],
|
||
|
'elevation': [],
|
||
|
'distance_to_water': []
|
||
|
}
|
||
|
|
||
|
for longitude in range(-179, 179):
|
||
|
print('-', end='')
|
||
|
for latitude in range(-89, 89):
|
||
|
data['longitude'].append(longitude)
|
||
|
data['latitude'].append(latitude)
|
||
|
|
||
|
elev = elevation_data[elevation.index(longitude, latitude)]
|
||
|
data['elevation'].append(elev)
|
||
|
data['distance_to_water'] = 999
|
||
|
print('+', end='')
|
||
|
|
||
|
print('')
|
||
|
|
||
|
df = pd.DataFrame(data)
|
||
|
|
||
|
predict_end_to_end(df)
|
||
|
plt.show()
|