生成复数的二维数组

时间:2016-08-28 11:00:14

标签: python numpy

我想生成一个包含复数的给定大小的二维数组,如下例所示:

title <- "almourol_gfdl-esm2m_rcp8p5_daily_ref.csv"
title <- strsplit(title, "_")[[1]]
title <- noquote(title)
title <- paste(title[1],title[4], "discharge",title[5], "period")
title

[1] "almourol daily discharge ref.csv period"

2 个答案:

答案 0 :(得分:2)

这是实现这一目标的一种方式 使用np.indices()参与np.vectorize()

def generate_array(dim):
    X, Y = np.indices(dim)
    return np.array(np.vectorize(complex)(X, Y))

ar = generate_array((2, 3))
print(ar)

输出:

[[ 0.+0.j  0.+1.j  0.+2.j]
 [ 1.+0.j  1.+1.j  1.+2.j]]

答案 1 :(得分:2)

class FooAdmin(admin.ModelAdmin):
     list_display = ('bar', 'get_is_new_bar') 

     def get_is_new_bar(self, obj):
        return obj.is_new_bar
     get_is_new_bar.boolean = True