- 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/
在开发中的网页上,我在 IE 上遇到此错误 element = $(element); 此代码位于prototype.js 预期对象 如何消除此错误。 更新: 现场也使用了 jQuery。 最佳答
我有两个大小相同的嵌套数组: Array1 =[[1, 2], [], [2, 3]] Array2= [[1, 4], [8, 11], [3, 6]] 我需要将它们合并到一个数组中,如下所示: A
我有一些 jQuery 代码,当单击具有特定 ID 的项目时运行。当 ID 是 的一部分时,它就可以工作。元素,但当它位于 中时则不然元素。为什么会这样呢?我想使用 an,因为如果用户关闭了 Ja
Flex-box 规范 3声明 flex 元素不是 block 容器: A flex item establishes a new formatting context for its content
我遇到了一个意想不到的问题。 HTML JS $(function() { var $divs = $('.myDiv'); // create new div not in
我使用 Bootstrap 和 Ember.js 得到了一个无序列表。每个列表项都是一个显示新帖子的链接,每当您单击该链接时,Ember 都会添加类 active默认情况下。我正在使用 Bootstr
我正在尝试让一个函数正常工作,但运气不佳,所以我想向 Stackoverflow 智囊团提出一个新手问题! 基本上,我有一个表单,并且循环遍历所有元素以查看是否存在自定义数据属性。如果存在,则保持该元
我想映射一个可选数组,删除那些 nil 值,并使用另一个函数映射非 nil 值。 我知道我可以通过使用 compactMap 然后使用常规 map 来实现这一点,但我只想遍历数组一次。 我为此实现了一
我如何定位 li 元素,除非它们出现在 之后元素?换句话说,我想针对步骤而不是注释。 我尝试向 OL 添加一个我想从选择中排除的类,但我想出的代码不起作用。 (顺便说一句,重构 html 不是一种选
Warning 1 The element 'system.webServer' has invalid child element 'rewrite'. List of possible eleme
我正在尝试编写一个脚本,该脚本将遍历 HTML 源并创建 DOM 的 JSON 文件,然后使用 d3.js 在 TreeView 中显示该文件。我遇到的问题是不仅希望显示元素(TITLE、P、LI 等
我有以下 HTML 表单:- Option 1 Option 2
我试图在选定的 HTML 元素之后选择下一个具有类名 slider-value 的 span 元素。我尝试了多种解决方案,但没有一个有效。 我可以通过 id 选择它,但我不希望那样做使代码冗余。 $(
如果电子邮件地址无效,我想在屏幕上显示一条消息“请输入有效的电子邮件地址”。 body 元素的innerHTML 语句工作正常,但我用于p 元素的innerHTML 语句不起作用。 有一次,当我测试它
以下 jQuery 代码调用 ul 元素,查找元素内的前 三个 li 列表项,并隐藏剩余的 li 项目。然后,它附加一个 li 元素,其中显示“显示更多...”,并且在单击时显示之前隐藏的列表项。 (
我问了a question早些时候关于将编辑/删除链接与 h1 元素内联的最佳方法。我能够通过给出的答案实现这一点,但我现在有额外的要求,我需要在 h1 下方显示一个段落并编辑/删除链接。 到目前为止
我使用 MVC 4 和 knockout.js 库版本 2.1.0 显示从服务器检索到的大量文件的表中的以下摘录。 0)"> 正在正确检索数据,
我创建了一个脚本,该脚本在鼠标悬停在父容器上时激活,并且应该将其子元素移离鼠标。我目前已经让它工作了,但是代码的某些部分似乎与 REACT 代码应该是什么样子相矛盾。特别是两个部分。 我在渲染函数中使
我是 JS 新手,正在尝试理解项目 https://github.com/tastejs/todomvc 的代码 请参阅屏幕截图,我尝试对 button X 以及其父元素 div 设置断点,但在这两种
例如,假设有一个带有奇特颜色的标记: Something written here 使用 Visual Studio 2017 和 MVC 5 元素,有没有办法检查和定位当前应用了哪些样式,以及负责它
我是一名优秀的程序员,十分优秀!