pydub_numpy.py

import numpy as np
import pydub

def audiosegment2array(segment):
  return np.array(segment.get_array_of_samples()).reshape(-1, segment.channels)

def array2audiosegment(array, rate=44100):
  assert array.ndim == 2
  num_channels = array.shape[1]
  array = bytes(array.astype(np.int16).reshape(-1).data)
  return pydub.AudioSegment(
    array,
    sample_width=2,
    frame_rate=rate,
    channels=num_channels
  )