首页 >>  正文

python求100以内的质数

来源:baiyundou.net   日期:2024-09-22

丰色 发自 凹非寺

量子位 | 公众号 QbitAI

众所周知,Python的简单和易读性是靠牺牲性能为代价的——

尤其是在计算密集的情况下,比如多重for循环。

不过现在,大佬胡渊鸣说了:

只需import 一个叫做“Taichi”的库,就可以把代码速度提升100倍

不信?

来看三个例子。

计算素数的个数,速度x120

第一个例子非常非常简单,求所有小于给定正整数N的素数。

标准答案如下:

我们将上面的代码保存,运行。

当N为100万时,需要2.235s得到结果:

现在,我们开始施魔法。

不用更改任何函数体,import“taichi”库,然后再加两个装饰器:

Bingo!同样的结果只要0.363s,快了将近6倍。

如果N=1000万,则只要0.8s;要知道,不加它可是55s,一下子又快了70倍

不止如此,我们还可以在ti.init()中加个参数变为ti.init(arch=ti.gpu) ,让taich在GPU上进行计算。

那么此时,计算所有小于1000万的素数就只耗时0.45s了,与原来的Python代码相比速度就提高了120倍

厉不厉害?

什么?你觉得这个例子太简单了,说服力不够?我们再来看一个稍微复杂一点的。

动态规划,速度x500

动态规划不用多说,作为一种优化算法,通过动态存储中间计算结果来减少计算时间。

我们以经典教材《算法导论》中的经典动态规划案例“最长公共子序列问题(LCS)”为例。

比如对于序列a = [0, 1, 0, 2, 4, 3, 1, 2, 1]和序列b = [4, 0, 1, 4, 5, 3, 1, 2],它们的LCS就是:

LCS(a, b) = [0, 1, 4, 3, 1, 2]。

用动态规划的思路计算LCS,就是先求解序列a的前i个元素和序列b的前j个元素的最长公共子序列的长度,然后逐步增加i或j的值,重复过程,得到结果。

我们用f[i, j]来指代这个子序列的长度,即LCS((prefix(a, i), prefix(b, j)。其中prefix(a, i) 表示序列a的前i个元素,即a[0], a[1], …, a[i - 1],得到如下递归关系:

完整代码如下:

现在,我们用Taichi来加速:

结果如下:

胡渊鸣电脑上的程序最快做到了0.9秒内完成,而换成用NumPy来实现,则需要476秒,差异达到了超500倍!

最后,我们再来一个不一样的例子。

反应 - 扩散方程,效果惊人

自然界中,总有一些动物身上长着一些看起来无序但实则并非完全随机的花纹。

图灵机的发明者艾伦·图灵是第一个提出模型来描述这种现象的人。

在该模型中,两种化学物质(U和V)来模拟图案的生成。这两者之间的关系类似于猎物和捕食者,它们自行移动并有交互:

  1. 最初,U和V随机分布在一个域上;
  2. 在每个时间步,它们逐渐扩散到邻近空间;
  3. 当U和V相遇时,一部分U被V吞噬。因此,V的浓度增加;
  4. 为了避免U被V根除,我们在每个时间步添加一定百分比 (f) 的U并删除一定百分比 (k) 的V。

上面这个过程被概述为“反应-扩散方程”:

其中有四个关键参数:Du(U的扩散速度),Dv(V的扩散速度),f(feed的缩写,控制U的加入)和k(kill的缩写,控制V的去除)

如果Taichi中实现这个方程,首先创建网格来表示域,用vec2表示每个网格中U, V的浓度值。

拉普拉斯算子数值的计算需要访问相邻网格。为了避免在同一循环中更新和读取数据,我们应该创建两个形状相同的网格W×H×2。

每次从一个网格访问数据时,我们将更新的数据写入另一个网格,然后切换下一个网格。那么数据结构设计就是这样:

一开始,我们将U在网格中的浓度设置为 1,并将V放置在50个随机选择的位置:

那么实际计算就可以用不到10行代码完成:

","force_purephv":"0","gnid":"937e299d4c2c2a176","img_data":[{"flag":2,"img":[{"desc":"","height":"920","title":"","url":"https://p0.ssl.img.360kuai.com/t01e23794c94540b7d3.jpg","width":"920"},{"desc":"","height":"765","title":"","url":"https://p0.ssl.img.360kuai.com/t0144fb37a5bf3f53de.jpg","width":"1080"},{"desc":"","height":"227","title":"","url":"https://p0.ssl.img.360kuai.com/t0150b2fef7620e3dbc.jpg","width":"1080"},{"desc":"","height":"839","title":"","url":"https://p0.ssl.img.360kuai.com/t016b17cfbc73cc0857.jpg","width":"1080"},{"desc":"","height":"218","title":"","url":"https://p0.ssl.img.360kuai.com/t0179edf8404f9ae37d.jpg","width":"1080"},{"desc":"","height":"106","title":"","url":"https://p0.ssl.img.360kuai.com/t01e32851aaad11b65f.jpg","width":"1080"},{"desc":"","height":"179","title":"","url":"https://p0.ssl.img.360kuai.com/t01bec382de36100f5c.jpg","width":"1080"},{"desc":"","height":"1048","title":"","url":"https://p0.ssl.img.360kuai.com/t01d29f206b72759e5c.jpg","width":"952"},{"desc":"","height":"220","title":"","url":"https://p0.ssl.img.360kuai.com/t013e42423d8b222dd8.jpg","width":"1080"},{"desc":"","height":"200","title":"","url":"https://p0.ssl.img.360kuai.com/t01d2b2fb1751d67f82.jpg","width":"1080"},{"desc":"","height":"290","title":"","url":"https://p0.ssl.img.360kuai.com/t01a6e79e099eafb7c4.jpg","width":"732"},{"desc":"","height":"103","title":"","url":"https://p0.ssl.img.360kuai.com/t0120a647a3f436ce6b.jpg","width":"1080"},{"desc":"","height":"321","title":"","url":"https://p0.ssl.img.360kuai.com/t01758422991b7876fd.jpg","width":"1080"}]}],"original":0,"pat":"art_src_3,fts0,sts0","powerby":"hbase","pub_time":1662700080000,"pure":"","rawurl":"http://zm.news.so.com/ff7cacee7feffebd5b82b9dcf2d60187","redirect":0,"rptid":"aec5ec457dee7f65","s":"t","src":"量子位","tag":[],"title":"胡渊鸣:import一个“太极”库,让Python代码提速100倍

徒钥怨4148用python3点85,写100以内三的倍数怎么写 -
嵇兔往15373054233 ______ for i in range(100): if i % 3 == 0: if i != 0: print(i)

徒钥怨4148怎么用Python编程实现输出100以内逢7拍手的数字的程序?
嵇兔往15373054233 ______ for i in range(1,100): x=i k=i//7 y=7*k if y==x: print('拍手',y)

徒钥怨4148求助python 查询2到100之间的素数: 代码: i =2 j =1 c=0 while -
嵇兔往15373054233 ______ 这是3-100的素数,供参考!def is_prime(n): i = 2 while(i < n): if n % i == 0: break i += 1 if n == i: return True else: return False p = 0 for i in range(3,101): if is_prime(i): print i, p += 1 if p % 10 == 0: print '\n'

徒钥怨4148使用Python编程,求1~100所有偶数的阶乘. -
嵇兔往15373054233 ______ total = 1 for num in range(1,101): if num % 2 == 0: total = total * num print(total) #其实抄一楼答案基本都对了,就是21135261range默认是从0开始的, #而0对2的求4102余也是0,所以答案永远是0,设置一个开始值为1,或者16532就行了

徒钥怨4148python 100以内素数的和 -
嵇兔往15373054233 ______ def sundaram3(max_n): numbers=range(3,max_n+1,2) half=(max_n)//2 initial=4 for step in xrange(3,max_n+1,2): for i in xrange(initial,half,step): numbers[i-1]=0 initial+=2*(step+1) if initial>half: return[2]+filter(None,numbers) print(sum(sundaram3(100)))

徒钥怨4148100以内10个最大素数求和代码 -
嵇兔往15373054233 ______ C可以这么写:#include "stdio.h" int prime(int n){ int i; for(i=3; i*i<=n;i+=2) if(!(n%i)) return 0; return 1; } int main(void){ int i,sum,n; for(sum=i=0,n=97;i<10;n-=2) if(prime(n)) sum+=n,i++; printf("The result is %d\n",sum); return 0; }

徒钥怨4148求python代码:输出100以内的勾股数,输出到一个文件中,看看我的要怎么处理,输出后老是有重复 -
嵇兔往15373054233 ______ for a in range(100): for b in range(a-1): for c in range(b-1): if a**2=b**2+c**2:python不熟悉,大概可以这么弄.

徒钥怨4148python编写程序:100到1000内10的任意整数并求出最大值 -
嵇兔往15373054233 ______ 如果只是产生最大值和最小值的话,只添加一个command就可以了. 之后写这些代码 . Private Sub Command1_Click() Randomize '产生随机数种子 以防止每次生成结果都一样 Dim a(1 To 100) As Integer '声明数组存放数据 用来保存随机数结果 ...

徒钥怨4148python循环素数数字197可以被称为循环素数,因为197的三个数位循环移位后的数字:197,971,719均为素数.100以内这样的数字包括13个,2,3,5,7,11,13,17,... -
嵇兔往15373054233 ______[答案] #!/usr/bin/python # encoding: utf-8 import time import pprint def allPrime(maxNum): aList = range(0, maxNum) def _do(aPrime): for i in xrange(aPrime, maxNum, aPrime): aList[i] = 0 return aPrime return map(_do, (i for i in xrange(2,maxNum) if aList[i]) ) t...

徒钥怨4148python 求教如何输出0 - 100内前5个能被3整除的数? -
嵇兔往15373054233 ______ j=1 for i in range(100): if j<6: if i % 3 == 0: print i j += 1

(编辑:自媒体)
关于我们 | 客户服务 | 服务条款 | 联系我们 | 免责声明 | 网站地图 @ 白云都 2024