using chatgpt to translate r code to python effed461037e