Python:规范化多维数组

时间:2014-04-24 12:44:27

标签: python numpy multidimensional-array

我有生成和导出12个立体声WAV的代码

目前我正在做:

for i in range(0,12):
    filename = "out_%d.wav" % i
    L = ... some list of floats between -1.0 and +1.0 ...
    R = ... similarly ...

    exportWAV(filename,L,R)

但音量过于安静。

我需要做的是找到所有L& amp;的最大音量。 R,并划分所有L& R由此卷。然后我的所有值都在-1和1之间。

这不是一项艰巨的任务,我可以破解一些丑陋的代码来完成它。

但如何干净利落呢?

我应该能够在几行内完成,例如:

all_LR = ''
all_file = ''

for i in range(0,12):
    filename = ...
    L, R = ...

    all_LR += (L,R)
    all_file += filename

maxVal = max( abs(all_LR) )
all_LR /= maxVal

for file,L,R in zip( all_file, all_LR ):
    exportWAV(filename L,R)

但我无法看到如何将这个伪代码转换为实际有效的Python。 abs和max不对元组数组进行操作,元组中的每个元素本身都是一个浮点数组。

我觉得我只是通过尝试保存几行代码来使它变得更加困难。

编辑:由于以下答案,我现在有以下工作代码:

all_LR = []

for i in range(0,12):
    print "Processing %d" % i

    hrtf_file = hrtf_path + "/%02d.wav" % (i+1)
    shep_file = shepard_path + "/shepard_%02d.WAV" % i

    L, R = import_WAV( hrtf_file )
    shep = import_WAV( shep_file )

    out_L = np.convolve(shep,L)
    out_R = np.convolve(shep,R)

    #out_LR = np.array( out_L, out_R )
    out_LR = (list(out_L), list(out_R))
    all_LR.append(out_LR)
    #np.append( all_LR, out_LR )

np_array = np.array(all_LR)

amp_max = np.amax( np.fabs(np_array) )
print( "AmpMAX: %f" % amp_max )

np_array /= amp_max

for i in range(0,12):
    out_file = out3d_path + "/s3D_%02d.WAV" % i
    print out_file

    L,R = np_array[i]

    export_WAV( out_file, L, R )

1 个答案:

答案 0 :(得分:3)

如果您可以将数据转换为numpy.arrays,那么您可以使用numpy.amaxnumpy.fabs,如下所示。

import numpy as np

a = np.array([(1, 2), (-3, 4), (-2, 2), (0, 1), (1, 3)])

a_abs = np.fabs(a)
print(a_abs)
# [[ 1.  2.]
#  [ 3.  4.]
#  [ 2.  2.]
#  [ 0.  1.]
#  [ 1.  3.]]

a_max = np.amax(a_abs)

print(a_max)
# 4.0

np.fabs将返回一个相同形状的数组,但是多维数组中每个元素的绝对值。

np.amax将返回数组的最大值。如果您使用关键字参数(例如axis)选择np.amax(a, axis=0),那么它将沿axis行动。但axis关键字的默认值为None,这将使其沿着展平数组行事。