根據自定義函數的bin寬度

Question

我想以非均勻的方式將我的數據打包。

import numpy as np 
import matplotlib.pyplot as plt 

def fun(x): 
    return >some function of x< 

我現在想的是使somehing看起來像：

np.linespace(0.,200.,fun(x)) 

是否有使用numpy的功能，可以處理它的便捷方式？ 我剛纔看到np.arange和np.linspace使用數字，不能將函數作爲參數。

我可以編寫一個可以處理它的函數，但本機解決方案會更好。

Answer 1

創建一個x值範圍，定義您的自定義函數，然後在每個數組元素上調用您的函數。 Numpy使這很容易：

fun = lambda x: x**2 # Example function 
N = 10 # Number of data points 
x = np.linspace(0., 200., N) # Creates an array of N points 
bins = fun(x) # Applies fun to all values in array x