使用 multiprocessing 或 concurrent.futures 将数据帧中的多列传递到函数中

如何解决使用 multiprocessing 或 concurrent.futures 将数据帧中的多列传递到函数中

问题:如何使用 multiprocessingconcurrent.futures 将数据框的列传递给每行的函数?

详情: 对于 df 中的每一行,我想将其列 leaderyears 传递给函数 print_sentences()。我想以并行方式使用该函数,其中每一行都是异步打印的。例如,我想使用 concurrent.futures.Executor.map

它需要在 Python 3.6.

Reprex:我的实际问题在计算上要求很高,所以这里是一个简化的 reprex

import pandas as pd
import numpy as np
import concurrent.futures

df = pd.DataFrame(np.array([["Larry",3,"Germany"],["Jerry",5,"Sweden"],["George",12,"UK"]]),columns=['leader','years','score'])

def print_sentences(df):

    print(df["leader"] + " has been leader for " + df["years"] + " years")


print_sentences(df)

背景: 与此问题相关的其他问题似乎与 dataframe 以外的对象类型有关。

当我读入数据帧 .csv 时,我的具体问题就开始了。我想将此数据框的每一行的列传递给某个函数。我的实际功能(对 reprex 进行了极大简化)在计算上要求很高。它抓取数据并将其保存到 .json。因此,每一行都充当不同的查询(例如,输入不同的领导者的姓名和分数)。

为了优化这一点,我希望行以并行方式映射到函数中。

我用上面的 reprex 简化了我的问题。

提前感谢您的帮助。

解决方法

试试这个,

#Edits 反映您的用例。

import pandas as pd
import numpy as np
from multiprocessing import cpu_count,Pool

cores = cpu_count() #Number of CPU cores on your system
partitions = cores #Define as many partitions as you want

def parallelize(data,func):
    data_split = np.array_split(data,partitions)
    pool = Pool(cores)
    data = pd.concat(pool.map(func,data_split))
    pool.close()
    pool.join()
    return data

def print_sentences(cols):
    leader,years = cols[0],cols[1]
    print(leader + " has been leader for " + years + " years")


df = pd.DataFrame(np.array([["Larry",3,"Germany"],["Jerry",5,"Sweden"],["George",12,"UK"]]),columns=['leader','years','score'])

data = df.copy()

data = parallelize(data,print_sentences)

data.apply(print_sentences,axis=1)

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。

相关推荐


Selenium Web驱动程序和Java。元素在(x,y)点处不可单击。其他元素将获得点击?
Python-如何使用点“。” 访问字典成员?
Java 字符串是不可变的。到底是什么意思?
Java中的“ final”关键字如何工作?(我仍然可以修改对象。)
“loop:”在Java代码中。这是什么,为什么要编译?
java.lang.ClassNotFoundException:sun.jdbc.odbc.JdbcOdbcDriver发生异常。为什么?
这是用Java进行XML解析的最佳库。
Java的PriorityQueue的内置迭代器不会以任何特定顺序遍历数据结构。为什么?
如何在Java中聆听按键时移动图像。
Java“Program to an interface”。这是什么意思?
Java在半透明框架/面板/组件上重新绘画。
Java“ Class.forName()”和“ Class.forName()。newInstance()”之间有什么区别?
在此环境中不提供编译器。也许是在JRE而不是JDK上运行?
Java用相同的方法在一个类中实现两个接口。哪种接口方法被覆盖?
Java 什么是Runtime.getRuntime()。totalMemory()和freeMemory()?
java.library.path中的java.lang.UnsatisfiedLinkError否*****。dll
JavaFX“位置是必需的。” 即使在同一包装中
Java 导入两个具有相同名称的类。怎么处理?
Java 是否应该在HttpServletResponse.getOutputStream()/。getWriter()上调用.close()?
Java RegEx元字符(。)和普通点?