python 读取excl内容当变量

读取同目录文件夹下excl内容当变量

def get_neg_domains(number=200):

df = pd.DataFrame(pd.read_csv("DGA_train.csv", names=["index", "domains", "target"]))

domains = df["domains"].tolist()

return domains[:number]

if __name__ == '__main__':

neg_domain_number = 2000

neg_domains = get_neg_domains(neg_domain_number)

domain1 = random.choice(neg_domains)

domain_2 = [str(j) for j in domain1]

domain = ''.join(domain_2)

domain为变量就获取完成了

同一路径下存在文件DGA_train.csv names为列名