我有这样的代码
def function1(idData):
movie = movdb.Movies(idData)
response = movie.info()
runtime = movie.runtime
return runtime
def function2(idData):
movie = movdb.Movies(idData)
response = movie.info()
lang = movie.original_language
return lang
def function2(idData):
movie = movdb.Movies(idData)
response = movie.info()
overview = movie.overview
return overview
def main():
df['Column 1'] = df.apply(lambda row : function1(row['Id'], axis = 1)
df['Column 2'] = df.apply(lambda row : function2(row['Id']), axis = 1)
df['Column 3'] = df.apply(lambda row : function3(row['Id']), axis = 1)
if __name__ == '__main__':
main()
这很有效,但我觉得效率不高。有没有办法返回多个值到多个列?所以我只需要访问下面的代码一次。
movie = tmdb.Movies(idData)
response = movie.info()
或者你有什么想法使这段代码更有效?
# # #
def function(idData):
movie = movdb.Movies(idData)
response = movie.info()
runtime = movie.runtime
lang = movie.original_language
overview = movie.overview
return [[runtime,lang,overview]]
def main():
df[['Column 1','Column 2','Column 3']] = df.apply(lambda row : function1(row['Id'], axis = 1)
if __name__ == '__main__':
main()