导入模块后如何修改函数内固定参数以复用关联函数？

Absolutely, there are several clean ways to achieve exactly what you want—letting you adjust the parameters inside function(i) without modifying plotme(), average(), or other dependent functions. Below are the most practical approaches tailored to your ideal workflow:

1. Make function(i) Configurable via Keyword Arguments (Matches Your Ideal Workflow) #

This approach lets you update parameters directly when calling function(i) (or even without passing an i if you just want to set params), and all subsequent calls to plotme()/average() will use the new values.

Modify your mymodule.py: #

import scipy.stats as stats

# Internal default parameters
_beta_a = 10
_beta_b = 2
_beta_scale = 100
_beta_loc = -50

def function(i=None, a=None, b=None, scale=None, loc=None):
    global _beta_a, _beta_b, _beta_scale, _beta_loc
    
    # Update parameters if new values are provided
    if a is not None:
        _beta_a = a
    if b is not None:
        _beta_b = b
    if scale is not None:
        _beta_scale = scale
    if loc is not None:
        _beta_loc = loc
    
    # Calculate and return the PDF value only if i is provided
    if i is not None:
        return stats.beta.pdf(i, a=_beta_a, b=_beta_b, scale=_beta_scale, loc=_beta_loc)

# Your existing dependent functions stay exactly the same
def plotme():
    # Example plotting logic (replace with your actual code)
    xs = range(-40, 60, 10)
    ys = [function(x) for x in xs]
    print(f"Plotting beta PDF values: {ys}")

def average():
    # Example average calculation
    xs = range(-40, 60, 10)
    avg = sum(function(x) for x in xs) / len(xs)
    print(f"Average of beta PDF: {avg}")

How to use it: #

import mymodule

# Option 1: Set parameters while calculating a value (matches your example)
mymodule.function(0, a=500, b=200, scale=50, loc=0)

# Option 2: Set parameters without calculating a value (cleaner)
# mymodule.function(a=500, b=200, scale=50, loc=0)

# Now plotme() and average() will use the new parameters
mymodule.plotme()
mymodule.average()

2. Use a Class to Encapsulate Parameters (More Object-Oriented) #

If you prefer a more structured approach (and want to support multiple parameter sets if needed), wrap the function and its parameters in a class. This keeps parameters encapsulated and avoids global variables.

Modify your mymodule.py: #

import scipy.stats as stats

class BetaPDF:
    def __init__(self, a=10, b=2, scale=100, loc=-50):
        self.a = a
        self.b = b
        self.scale = scale
        self.loc = loc
    
    def __call__(self, i):
        return stats.beta.pdf(i, a=self.a, b=self.b, scale=self.scale, loc=self.loc)

# Create a default instance for plotme() and average() to use
function = BetaPDF()

# Dependent functions remain unchanged
def plotme():
    xs = range(-40, 60, 10)
    ys = [function(x) for x in xs]
    print(f"Plotting beta PDF values: {ys}")

def average():
    xs = range(-40, 60, 10)
    avg = sum(function(x) for x in xs) / len(xs)
    print(f"Average of beta PDF: {avg}")

How to use it: #

import mymodule

# Option 1: Replace the default instance with a new one
mymodule.function = mymodule.BetaPDF(a=500, b=200, scale=50, loc=0)

# Option 2: Modify individual parameters of the existing instance
# mymodule.function.a = 500
# mymodule.function.b = 200
# mymodule.function.scale = 50
# mymodule.function.loc = 0

mymodule.plotme()
mymodule.average()

3. Module-Level Parameters (Simplest Minimal Change) #

If you want the least amount of code modification, define the parameters as module-level variables that function(i) references. You can then modify these variables directly after importing the module.

Modify your mymodule.py: #

import scipy.stats as stats

# Exposed module-level parameters (default values)
beta_a = 10
beta_b = 2
beta_scale = 100
beta_loc = -50

def function(i):
    return stats.beta.pdf(i, a=beta_a, b=beta_b, scale=beta_scale, loc=beta_loc)

# Dependent functions stay the same
def plotme():
    xs = range(-40, 60, 10)
    ys = [function(x) for x in xs]
    print(f"Plotting beta PDF values: {ys}")

def average():
    xs = range(-40, 60, 10)
    avg = sum(function(x) for x in xs) / len(xs)
    print(f"Average of beta PDF: {avg}")

How to use it: #

import mymodule

# Update the module-level parameters directly
mymodule.beta_a = 500
mymodule.beta_b = 200
mymodule.beta_scale = 50
mymodule.beta_loc = 0

mymodule.plotme()
mymodule.average()

Which to Choose? #

• Option 1: Best if you want to stick exactly to your ideal workflow (setting params via function() calls).
• Option 2: Great if you need flexibility (e.g., multiple beta PDF configurations at once) or prefer OOP practices.
• Option 3: The simplest fix when you just want minimal code changes.

All these methods let you avoid modifying plotme(), average(), or other functions that rely on function(i)—they’ll automatically use the updated parameters.

内容的提问来源于stack exchange，提问作者Tylerr