streamlit_leisure 진행중 에러(3) - ValueError: Lengths must match to compare

https://github.com/reodinas/streamlit_leisure

프로젝트 진행 중 발생한 문제 해결 과정을 기록한 포스트입니다.

 

 

유저에게 선택을 받아 선택받은 데이터만을 보여주는 기능을 구현하다 또 이 에러가 발생했다.

주피터 노트북에서 테스트 할 때는 잘 됐었기 때문에 잠시 헤맸는데,

결론은 박스 친 코드가 문제였다.

 

 

selected group2가 하나의 문자열일 때는 문제가 없었지만,

st.multiselect로 입력 받은 값은 리스트가 되기 때문에 

문자열이 리스트와 같냐고 물어본 꼴이다. 따라서 짝이 맞지 않다고 에러가 뜬다.

 

이 버그를 수정하기 위해 박스 친 부분의 코드를 apply 함수를 사용해 다음과 같이 바꿔주었다.

df_new = df[df[selected_group].apply(lambda x: x in selected_group2)][selected_col]