Как изменить ограничения для geo_shape в altair (python vega-lite) - PullRequest
Новая
       10

Как изменить ограничения для geo_shape в altair (python vega-lite)

0 голосов
/ 29 января 2019

Я пытаюсь построить локации в трех штатах США на питоне с Альтаиром.Я видел учебник о карте США, но мне интересно, есть ли способ увеличить изображение до трех интересующих его состояний, то есть Нью-Йорка, Нью-Джерси и Коннектикута.

В настоящее время у меня есть следующий код: 

from vega_datasets import data            
states = alt.topo_feature(data.us_10m.url, 'states')

# US states background
background = alt.Chart(states).mark_geoshape(
    fill='lightgray',
    stroke='white',
    limit=1000
).properties(
    title='US State Capitols',
    width=700,
    height=400
).project("albers")

points=alt.Chart(accts).mark_point().encode(
        longitude = "longitude",
        latitude = "latitude",
        color = "Group")

background+points

Я проверил набор данных us_10m.url и похоже, что нет поля, которое указывает отдельные состояния.Поэтому я надеюсь, что смогу просто как-то изменить xlim и ylim для фона, например, на [-80, -70] и [35,45].Я хочу приблизить регионы, где есть точки данных (синие точки).

Может ли кто-нибудь показать мне, как это сделать?Спасибо !!

enter image description here

Обновление

В файле JSON есть поле с именем ID, и я вручную обнаружил, что NJ равен 34Нью-Йорк - 36, а CT - 9. Есть ли способ отфильтровать эти идентификаторы?Это сделает работу!

1 Ответ

0 голосов
/ 29 января 2019

Хорошо, похоже, что функция выделения / масштабирования / xlim / ylim для геотипа еще не поддерживается: Документируйте и добавьте предупреждение о том, что геоположение еще не поддерживает выбор # 3305

Таким образом, я получил хакерский способ решить эту проблему, сначала фильтруя по идентификаторам, используя чистый python.В основном, загрузите файл JSON в словарь, а затем измените поле значения перед преобразованием словаря в формат topojson.Ниже приведен пример для 5 штатов: PA, NJ, NY, CT, RI и MA. 

import altair as alt
from vega_datasets import data
# Load the data, which is loaded as a dict object
us_10m  = data.us_10m()
# Select the geometries under states under objects, filter on id (9,25,34,36,42,44)
us_10m['objects']['states']['geometries']=[item for item in us_10m['objects'] \
      ['states']['geometries'] if item['id'] in [9,25,34,36,42,44]]
# Make the topojson data
states = alt.Data(
    values=us_10m, 
    format=alt.TopoDataFormat(feature='states',type='topojson'))

# Plot background (now only has 5 states) 
background = alt.Chart(states).mark_geoshape(
    fill='lightgray',
    stroke='white',
    limit=1000
).properties(
    title='US State Capitols',
    width=700,
    height=400
).project("mercator")

# Plot the points
points=alt.Chart(accts).mark_circle(size=60).encode(
    longitude = "longitude",
    latitude = "latitude",
    color = "Group").project("mercator")

# Overlay the two plots
background+points

Полученный график выглядит нормально:

enter image description here

...