将当前代码应用于另一个数据框

时间:2019-11-16 14:15:00

标签: python pandas loops vectorization apply

以下是我当前的代码,如何将相同的代码应用于另一个数据框(导入另一个Excel工作表并对其应用以下相同的代码)

import pandas as pd
import numpy as np
book= pd.read_csv("book.csv")
book=book.dropna()
book.sort_values("Market Cap", ascending=False, inplace=True)
book = book.head(500)
book["EY"] = (book["Earnings Per Share LTM"] / book["Start Price"])
book["ROIC"] = (book["EBIT"]/ book["Invested Capital"])
book["Price Change"] =((book["Close Price"] - book["Start Price"]) / book["Start Price"])
book['EY Rank'] = book['EY'].rank(ascending=False)
book['ROIC rank'] = book['ROIC'].rank(ascending=False)

1 个答案:

答案 0 :(得分:0)

您可以定义一个函数并根据需要多次调用它:

import pandas as pd
import numpy as np

def some_func(file_name):
    book= pd.read_csv(file_name)
    book=book.dropna()
    book.sort_values("Market Cap", ascending=False, inplace=True)
    book = book.head(500)
    book["EY"] = (book["Earnings Per Share LTM"] / book["Start Price"])
    book["ROIC"] = (book["EBIT"]/ book["Invested Capital"])
    book["Price Change"] =((book["Close Price"] - book["Start Price"]) / book["Start Price"])
    book['EY Rank'] = book['EY'].rank(ascending=False)
    book['ROIC rank'] = book['ROIC'].rank(ascending=False)
    return book

file_name = "book.csv"
some_func(file_name)
file_name = "next_book.csv"
some_func(file_name)

这是您要寻找的吗?