[Python] Altair에서 tick label에 단위를 추가하자
12 Jan 2023들어가며
Python에서 차트를 그리기 위해 Altair 라이브러리를 사용하는데, 차트의 tick에 단위를 추가할 일이 생겼다. 그래서 이곳 저곳 찾아보았다.
label에 단위 추가하기
단위를 추가하는 법을 알아보기 위해 먼저 Altair 공식 페이지의 다음 코드를 사용하였다.
import altair as alt
import pandas as pd
source = pd.DataFrame({
'a': ['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I'],
'b': [28, 55, 43, 91, 81, 53, 19, 87, 52]
})
alt.Chart(source).mark_bar().encode(
x='a',
y='b'
)
y label에 단위를 추가하려면 다음과 같이 altair.Y의 labelExpr 파라미터에 "datum.value + '단위'" 식으로 넣으면 된다고 한다. 단위로 ‘개’를 추가한 모습이다.
...
y=alt.Y("b", axis=alt.Axis(labelExpr="datum.value + '개'"))
...
이렇게 직접적으로 문자열을 파라미터로 넣어주어 단위를 추가할 수 있었다. 그런데 나는 변수를 이용하여 단위를 추가하고 싶어서 다음과 같은 코드를 사용하였다.
...
unit = "개"
alt.Chart(source).mark_bar().encode(
x='a',
y=alt.Y("b", axis=alt.Axis(labelExpr="datum.value"+unit)
)
...
그런데 이게 웬걸? 갑자기 label이 없어졌다. 그래서 다르게 해봤다.
...
unit = "개"
alt.Chart(source).mark_bar().encode(
x='a',
y=alt.Y("b", axis=alt.Axis(labelExpr="datum.value + unit"))
)
...
그랬더니 이제는 아얘 실행조차 되지 않았다. 그래서 나는 다음과 같이 성공했을 때 문자열을 따옴표까지 똑같이 해서 넣어야 겠다고 생각했다.
...
unit = "m/s"
unitLabel = "datum.value + '" + unit + "'"
alt.Chart(source).mark_bar().encode(
x='a',
y=alt.Y("b", axis=alt.Axis(labelExpr=unitLabel))
)
...
그랬더니 추가한 단위가 잘 출력되는 것을 확인하였다. 변수로 단위를 넣고 싶을 때에는 위와 같은 형식으로 넣으면 될 것이다.
마치며
왜 저렇게 들어가야 되는지는 잘 모르겠다. 자바스크립트 에러가 나오는 것으로 보아 자바스크립트와 관련된 것 같은데… 아무튼 성공해서 다행이다.