- html - 出于某种原因,IE8 对我的 Sass 文件中继承的 html5 CSS 不友好?
- JMeter 在响应断言中使用 span 标签的问题
- html - 在 :hover and :active? 上具有不同效果的 CSS 动画
- html - 相对于居中的 html 内容固定的 CSS 重复背景?
这个问题与我几个月前的一个问题类似:Generating a numpy array with all combinations of numbers that sum to less than a given number 。在这个问题中,我想生成所有数字,其总和最多为一个常数,前提是每个元素都有一定的最大值。
这次我想计算总和恰好等于该常数的所有排列。这可以被视为计算整数分区的唯一排列,其中每个元素都有特定的最大值。最终结果应存储在 numpy 数组中。
使用生成器,一个衬垫即可实现我们想要的:
import numpy as np
from itertools import product
K = 3
maxRange = np.array([1,3,2])
states = np.array([i for i in product(*(range(i+1) for i in maxRange)) if sum(i)==K])
给予
array([[0, 1, 2],
[0, 2, 1],
[0, 3, 0],
[1, 0, 2],
[1, 1, 1],
[1, 2, 0]])
当 K=20
和 maxRange = [20]*6
时,我的性能非常慢。排列数量限制为 53130,但已经需要 20 秒。我的直觉告诉我这应该花费不到一秒钟的时间。
有人有更快的解决方案吗?我无法修改我之前问题的解决方案来解决这个问题,因为我不知道如何切断不再可能精确加起来为 K
的排列。
我不介意使用 numba 中的 @jit
运算符的解决方案...只要它们比我现在的速度更快!
提前致谢。
最佳答案
我不得不考虑这个问题很长时间,但我已经设法将解决方案修改为 Generating a numpy array with all combinations of numbers that sum to less than a given number对于这个问题:
对于分区数量,我们的想法是计算数组feasible_range
,该数组指定在某个阶段我们至少总共需要多少才能仍然达到max_sum
。例如,如果我们想要达到总共 3 个且 max_range[0] == 1
,那么在开始最后一个元素之前我们需要至少有 2 个。该数组由累积和得出:
feasible_range = np.maximum(max_sum - np.append(np.array([0]),np.cumsum(max_range)[:-1]),0)
现在我们可以像以前一样计算分区数,将永远无法导致可行分区的元素设置为 0。
def number_of_partitions(max_range, max_sum):
M = max_sum + 1
N = len(max_range)
arr = np.zeros(shape=(M,N), dtype = int)
feasible_range = max_sum - np.append(np.array([0]),np.cumsum(max_range)[:-1])
feasible_range = np.maximum(feasible_range,0)
arr[:,-1] = np.where(np.arange(M) <= min(max_range[-1], max_sum), 1, 0)
arr[:feasible_range[-1],-1] = 0
for i in range(N-2,-1,-1):
for j in range(max_range[i]+1):
arr[j:,i] += arr[:M-j,i+1]
arr[:feasible_range[i],i]=0 #Set options that will never add up to max_sum at 0.
return arr.sum(axis = 0),feasible_range
配分函数也有与之前类似的解释。
def partition(max_range, max_sum, out = None, n_part = None,feasible_range=None):
#Gives all possible partitions of the sets 0,...,max_range[i] that sum up to max_sum.
if out is None:
max_range = np.asarray(max_range, dtype = int).ravel()
n_part,feasible_range = number_of_partitions(max_range, max_sum)
out = np.zeros(shape = (n_part[0], max_range.size), dtype = int)
if(max_range.size == 1):
out[:] = np.arange(feasible_range[0],min(max_range[0],max_sum) + 1, dtype = int).reshape(-1,1)
return out
#Copy is needed since otherwise we overwrite some values of P.
P = partition(max_range[1:], max_sum, out=out[:n_part[1],1:], n_part = n_part[1:],feasible_range=feasible_range[1:]).copy()
S = max_sum - P.sum(axis = 1) #The remaining space in the partition
offset, sz = 0, 0
for i in range(max_range[0]+1):
#select indices for which there is remaining space
#do this only if adding i brings us within the feasible_range.
ind, = np.where(np.logical_and(S-i>=0,S-i <= max_sum-feasible_range[0]))
offset, sz = offset + sz, ind.size
out[offset:offset+sz, 0] = i
out[offset:offset+sz, 1:] = P[ind]
return out
对于 K=20
和 maxRange = [20]*6
,partition(maxRange,K)
需要 13 毫秒,而之前需要 18.5 秒。
我不太喜欢我必须复制的部分;通过颠倒顺序也许可以避免这种情况。不过现在速度已经足够好了。
关于python - 固定长度整数分区的唯一排列,其中每个元素都有一个最大值,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/39466720/
将 KLV 字符串拆分为键、长度、值作为元素的列表/元组的更有效方法是什么? 为了添加一点背景,前 3 位数字作为键,接下来的 2 位表示值的长度。 我已经能够使用以下代码解决该问题。但我不认为我的代
首先,我试图从文件中提取视频持续时间,然后在无需实际上传文件的情况下显示它。 当用户选择视频时 - 信息将显示在其下方,包括文件名、文件大小、文件类型。不管我的技能多么糟糕 - 我无法显示持续时间。我
我是 Scala 编程新手,这是我的问题:如何计算每行的字符串数量?我的数据框由一列 Array[String] 类型组成。 friendsDF: org.apache.spark.sql.DataF
我有一个React Web应用程序(create-react-app),该应用程序使用react-hook-forms上传歌曲并使用axios将其发送到我的Node / express服务器。 我想确
如果给你一个网络掩码(例如 255.255.255.0),你如何在 Java 中获得它的长度/位(例如 8)? 最佳答案 如果您想找出整数低端有多少个零位,请尝试 Integer.numberOfTr
我需要使用 jQuery 获取 div 数量的长度。 我可以得到它,但在两个单击事件中声明变量,但这似乎是错误的,然后我还需要使用它来根据数字显示隐藏按钮。我觉得我不必将代码加倍。 在这里摆弄 htt
我对此感到非常绝望,到目前为止我在 www 上找不到任何东西。 情况如下: 我正在使用 Python。 我有 3 个数组:x 坐标、y 坐标和半径。 我想使用给定的 x 和 y 坐标创建散点图。 到目
我有一个表单,我通过 jQuery 的加载函数动态添加新的输入和选择元素。有时加载的元素故意为空,在这种情况下我想隐藏容器 div,这样它就不会破坏样式。 问题是,我似乎无法计算加载的元素,因此不知道
我决定通过替换来使我的代码更清晰 if (wrappedSet.length > 0) 类似 if (wrappedSet.exists()) 是否有任何 native jq 函数可以实现此目的?或者
简单的问题。如果我有一个如下表: CREATE TABLE `exampletable` ( `id` int(11) NOT NULL AUTO_INCREMENT, `textfield`
我正在使用经典 ASP/MySQL 将长用户输入插入到我的数据库中,该输入是从富文本编辑器生成的。该列设置为 LONG-TEXT。 作为参数化查询(准备语句)的新手,我不确定用于此特定查询的数据长度。
我正在获取 Stripe 交易费用的值(value)并通过禁用的文本字段显示它。 由于输入文本域,句子出现较大空隙 This is the amount $3.50____________that n
我有一个 div,其背景图像的大小设置为包含。但是,图像是视网膜计算机(Macbook Pro 等)的双分辨率图像,所以我希望能够以某种方式让页面知道即使我说的是背景大小:包含 200x200 图像,
我正在开发一个具有“已保存”和“已完成”模块的小部件。当我删除元素时,它会从 dom 中删除/淡化它,但是当我将其标记为完成时,它会将其克隆到已完成的选项卡。这工作很棒,但顶部括号内的数字不适合我。这
我有一个来自 json 提要的数组,我知道在 jArray 中有一个联盟,但我需要计算出该数组的计数,以防稍后将第二个添加到提要中。目前 log cat 没有注销“teamFeedStructure”
目标:给定一个混合类型的数组,确定每个级别的元素数量。如果同一层有两个子数组,则它们的每个元素都计入该层元素的总数。 方法: Array.prototype.elementsAtLevels = fu
我需要帮助为 Java 中的单链表制作 int size(); 方法。 这是我目前所拥有的,但它没有返回正确的列表大小。 public int size() { int size = 0;
我正在为学校作业创建一个文件服务器应用程序。我目前拥有的是一个简单的 Client 类,它通过 TCP 发送图像,还有一个 Server 类接收图像并将其写入文件。 这是我的客户端代码 import
我有这对功能 (,) length :: Foldable t => t a -> b -> (Int, b) 和, head :: [a] -> a 我想了解的类型 (,) length he
我正在GitHub Pages上使用Jekyll来构建博客,并希望获得传递给YAML前题中Liquid模板的page.title字符串的长度,该字符串在每个帖子的YAML主题中。我还没有找到一种简单的
我是一名优秀的程序员,十分优秀!